├── README.md ├── assets └── representative │ └── afhq │ ├── ref │ ├── cat │ │ ├── flickr_cat_000495.jpg │ │ ├── flickr_cat_000557.jpg │ │ ├── pixabay_cat_000355.jpg │ │ ├── pixabay_cat_000491.jpg │ │ ├── pixabay_cat_000535.jpg │ │ ├── pixabay_cat_000623.jpg │ │ ├── pixabay_cat_000730.jpg │ │ ├── pixabay_cat_001479.jpg │ │ ├── pixabay_cat_001699.jpg │ │ └── pixabay_cat_003046.jpg │ ├── dog │ │ ├── flickr_dog_001072.jpg │ │ ├── pixabay_dog_000121.jpg │ │ ├── pixabay_dog_000322.jpg │ │ ├── pixabay_dog_000357.jpg │ │ ├── pixabay_dog_000409.jpg │ │ ├── pixabay_dog_000799.jpg │ │ ├── pixabay_dog_000890.jpg │ │ └── pixabay_dog_001082.jpg │ └── wild │ │ ├── flickr_wild_000731.jpg │ │ ├── flickr_wild_001223.jpg │ │ ├── flickr_wild_002020.jpg │ │ ├── flickr_wild_002092.jpg │ │ ├── flickr_wild_002933.jpg │ │ ├── flickr_wild_003137.jpg │ │ ├── flickr_wild_003355.jpg │ │ ├── flickr_wild_003796.jpg │ │ ├── flickr_wild_003969.jpg │ │ └── pixabay_wild_000637.jpg │ └── src │ ├── cat │ ├── flickr_cat_000253.jpg │ ├── pixabay_cat_000181.jpg │ ├── pixabay_cat_000241.jpg │ ├── pixabay_cat_000276.jpg │ └── pixabay_cat_004826.jpg │ ├── dog │ ├── flickr_dog_000094.jpg │ ├── pixabay_dog_000321.jpg │ ├── pixabay_dog_000322.jpg │ ├── pixabay_dog_001082.jpg │ └── pixabay_dog_002066.jpg │ └── wild │ ├── flickr_wild_000432.jpg │ ├── flickr_wild_000814.jpg │ ├── flickr_wild_002036.jpg │ ├── flickr_wild_002159.jpg │ └── pixabay_wild_000558.jpg ├── core ├── __init__.py ├── checkpoint.py ├── data_loader.py ├── model.py ├── solver_remix.py ├── utils.py └── wing.py ├── eval.sh ├── main.py ├── metrics ├── __init__.py ├── eval.py ├── fid.py ├── lpips.py └── lpips_weights.ckpt └── train.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/README.md -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/flickr_cat_000495.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/flickr_cat_000495.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/flickr_cat_000557.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/flickr_cat_000557.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_000355.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_000355.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_000491.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_000491.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_000535.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_000535.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_000623.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_000623.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_000730.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_000730.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_001479.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_001479.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_001699.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_001699.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/cat/pixabay_cat_003046.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/cat/pixabay_cat_003046.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/flickr_dog_001072.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/flickr_dog_001072.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000121.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000121.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000322.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000322.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000357.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000357.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000409.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000409.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000799.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000799.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_000890.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_000890.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/dog/pixabay_dog_001082.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/dog/pixabay_dog_001082.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_000731.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_000731.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_001223.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_001223.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_002020.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_002020.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_002092.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_002092.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_002933.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_002933.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_003137.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_003137.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_003355.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_003355.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_003796.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_003796.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/flickr_wild_003969.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/flickr_wild_003969.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/ref/wild/pixabay_wild_000637.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/ref/wild/pixabay_wild_000637.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/cat/flickr_cat_000253.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/cat/flickr_cat_000253.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/cat/pixabay_cat_000181.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/cat/pixabay_cat_000181.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/cat/pixabay_cat_000241.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/cat/pixabay_cat_000241.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/cat/pixabay_cat_000276.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/cat/pixabay_cat_000276.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/cat/pixabay_cat_004826.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/cat/pixabay_cat_004826.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/dog/flickr_dog_000094.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/dog/flickr_dog_000094.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/dog/pixabay_dog_000321.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/dog/pixabay_dog_000321.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/dog/pixabay_dog_000322.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/dog/pixabay_dog_000322.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/dog/pixabay_dog_001082.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/dog/pixabay_dog_001082.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/dog/pixabay_dog_002066.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/dog/pixabay_dog_002066.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/wild/flickr_wild_000432.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/wild/flickr_wild_000432.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/wild/flickr_wild_000814.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/wild/flickr_wild_000814.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/wild/flickr_wild_002036.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/wild/flickr_wild_002036.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/wild/flickr_wild_002159.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/wild/flickr_wild_002159.jpg -------------------------------------------------------------------------------- /assets/representative/afhq/src/wild/pixabay_wild_000558.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/assets/representative/afhq/src/wild/pixabay_wild_000558.jpg -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/checkpoint.py -------------------------------------------------------------------------------- /core/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/data_loader.py -------------------------------------------------------------------------------- /core/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/model.py -------------------------------------------------------------------------------- /core/solver_remix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/solver_remix.py -------------------------------------------------------------------------------- /core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/utils.py -------------------------------------------------------------------------------- /core/wing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/core/wing.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/eval.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/main.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/metrics/eval.py -------------------------------------------------------------------------------- /metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/metrics/fid.py -------------------------------------------------------------------------------- /metrics/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/metrics/lpips.py -------------------------------------------------------------------------------- /metrics/lpips_weights.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/metrics/lpips_weights.ckpt -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjf8899/ReMix_Pytorch/HEAD/train.sh --------------------------------------------------------------------------------