levihsu / OOTDiffusion

Official implementation of OOTDiffusion: Outfitting Fusion based Latent Diffusion for Controllable Virtual Try-on
Other
5.07k stars 740 forks source link

How to Create Cloth Mask, Image-Parse, OpenPose-Image, and OpenPose-JSON for Fine-Tuning Model with Custom Dataset? #184

Open Jaynls opened 1 month ago

Jaynls commented 1 month ago

Hello,

I am working on fine-tuning a model with my own dataset and need some guidance on the process. Specifically, I have 10 model images and 20 clothes images. I need to know:

Script/Files for Preprocessing:

Which script or file should I run to generate the required cloth-mask, human parse, and OpenPose images and JSON data from my dataset? Directory Structure for Training:

After generating the required preprocessing files, in which directory should I place them for training? Location of Fine-Tuned Model:

Once the training is complete, where can I find the fine-tuned model? Testing with Custom Data:

For testing the model with my own data, which checkpoint model file needs to be replaced or used? Any detailed instructions or pointers to relevant documentation would be highly appreciated!

Thank you in advance for your help!

parxivaal commented 1 week ago

Check this out Preprocessing!