├── .gitattributes ├── .github └── workflows │ └── main.yml ├── .gitignore ├── .gradio └── certificate.pem ├── LICENSE ├── LICENSE-MODEL ├── README.md ├── TrainingNotes.md ├── accelerate_config.yaml ├── app.py ├── app_lora.py ├── app_no_lora.py ├── cog.yaml ├── example ├── garment │ ├── 00034_00.jpg │ ├── 00035_00.jpg │ ├── 00055_00.jpg │ ├── 00057_00.jpg │ ├── 00064_00.jpg │ ├── 00067_00.jpg │ ├── 00069_00.jpg │ ├── 00396_00.jpg │ └── 04564_00.jpg ├── github.jpg ├── person │ ├── 00008_00.jpg │ ├── 00008_00_mask.png │ ├── 00055_00.jpg │ ├── 00055_00_mask.png │ ├── 00057_00.jpg │ ├── 00057_00_mask.png │ ├── 00064_00.jpg │ ├── 00064_00_mask.png │ ├── 00067_00.jpg │ ├── 00067_00_mask.png │ ├── 00069_00.jpg │ ├── 00069_00_mask.png │ ├── 1.jpg │ └── 1_mask.png ├── result │ ├── 1.png │ ├── 2.png │ └── 3.png └── tryoff_result │ ├── restored_garment1.png │ ├── restored_garment2.png │ ├── restored_garment3.png │ ├── restored_garment4.png │ ├── restored_garment5.png │ └── restored_garment6.png ├── image_datasets └── cp_dataset.py ├── paser_helper.py ├── predict.py ├── requirements.txt ├── script └── fid_eval.py ├── src └── flux │ └── train_utils.py ├── test_lora.png ├── train_flux_inpaint.py ├── train_flux_inpaint.sh ├── tryoff.sh ├── tryoff_inference.py ├── tryon.sh ├── tryon_inference.py └── tryon_inference_lora.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/main.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/.github/workflows/main.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/.gitignore -------------------------------------------------------------------------------- /.gradio/certificate.pem: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/.gradio/certificate.pem -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE-MODEL: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/LICENSE-MODEL -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/README.md -------------------------------------------------------------------------------- /TrainingNotes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/TrainingNotes.md -------------------------------------------------------------------------------- /accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/accelerate_config.yaml -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/app.py -------------------------------------------------------------------------------- /app_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/app_lora.py -------------------------------------------------------------------------------- /app_no_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/app_no_lora.py -------------------------------------------------------------------------------- /cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/cog.yaml -------------------------------------------------------------------------------- /example/garment/00034_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00034_00.jpg -------------------------------------------------------------------------------- /example/garment/00035_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00035_00.jpg -------------------------------------------------------------------------------- /example/garment/00055_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00055_00.jpg -------------------------------------------------------------------------------- /example/garment/00057_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00057_00.jpg -------------------------------------------------------------------------------- /example/garment/00064_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00064_00.jpg -------------------------------------------------------------------------------- /example/garment/00067_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00067_00.jpg -------------------------------------------------------------------------------- /example/garment/00069_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00069_00.jpg -------------------------------------------------------------------------------- /example/garment/00396_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/00396_00.jpg -------------------------------------------------------------------------------- /example/garment/04564_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/garment/04564_00.jpg -------------------------------------------------------------------------------- /example/github.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/github.jpg -------------------------------------------------------------------------------- /example/person/00008_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00008_00.jpg -------------------------------------------------------------------------------- /example/person/00008_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00008_00_mask.png -------------------------------------------------------------------------------- /example/person/00055_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00055_00.jpg -------------------------------------------------------------------------------- /example/person/00055_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00055_00_mask.png -------------------------------------------------------------------------------- /example/person/00057_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00057_00.jpg -------------------------------------------------------------------------------- /example/person/00057_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00057_00_mask.png -------------------------------------------------------------------------------- /example/person/00064_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00064_00.jpg -------------------------------------------------------------------------------- /example/person/00064_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00064_00_mask.png -------------------------------------------------------------------------------- /example/person/00067_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00067_00.jpg -------------------------------------------------------------------------------- /example/person/00067_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00067_00_mask.png -------------------------------------------------------------------------------- /example/person/00069_00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00069_00.jpg -------------------------------------------------------------------------------- /example/person/00069_00_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/00069_00_mask.png -------------------------------------------------------------------------------- /example/person/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/1.jpg -------------------------------------------------------------------------------- /example/person/1_mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/person/1_mask.png -------------------------------------------------------------------------------- /example/result/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/result/1.png -------------------------------------------------------------------------------- /example/result/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/result/2.png -------------------------------------------------------------------------------- /example/result/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/result/3.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment1.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment2.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment3.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment4.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment5.png -------------------------------------------------------------------------------- /example/tryoff_result/restored_garment6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/example/tryoff_result/restored_garment6.png -------------------------------------------------------------------------------- /image_datasets/cp_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/image_datasets/cp_dataset.py -------------------------------------------------------------------------------- /paser_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/paser_helper.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/fid_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/script/fid_eval.py -------------------------------------------------------------------------------- /src/flux/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/src/flux/train_utils.py -------------------------------------------------------------------------------- /test_lora.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/test_lora.png -------------------------------------------------------------------------------- /train_flux_inpaint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/train_flux_inpaint.py -------------------------------------------------------------------------------- /train_flux_inpaint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/train_flux_inpaint.sh -------------------------------------------------------------------------------- /tryoff.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/tryoff.sh -------------------------------------------------------------------------------- /tryoff_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/tryoff_inference.py -------------------------------------------------------------------------------- /tryon.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/tryon.sh -------------------------------------------------------------------------------- /tryon_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/tryon_inference.py -------------------------------------------------------------------------------- /tryon_inference_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nftblackmagic/catvton-flux/HEAD/tryon_inference_lora.py --------------------------------------------------------------------------------