TryOnGAN-pytorch

WIP : Unofficial pytorch implementation of TryOnGAN There are multiple branches :

• UC : Unconditioned StyleGAN2-ada, without any changes.
• PC-add : Pose conditioned with pose encoders outputs added to style block outputs.
• PC-concat : Pose conditioned with pose encoder's outputs concatenated to style block outputs.

Useful Notebooks :

• Alphapose notebook to create pose keypoints
• Training Notebook for UC :
• Training Notebook for PC :

Useful links (Drive) :

• Trained Checkpoints
• Image Dataset
• Pose keypoints

Some interesting results:

• Interpolation

Color	Length	Region

I inverted two real images into latent codes and interpolated between the two codes. Interestingly the intermediate images are valid garments. Maybe we can extend this technique to generate novel garments composed of multiple components. Note that pose changes for UC models but not with PC. Scratch here means model was trained with random weights without transfer learning.

• Style mixing for UC model

A is images generated by mixing latents of a real image with random latent code. B is by mixing two real images. Mixing here means replacing latent codes for lower six style blocks with other latent codes.

Datasets

Some of these datasets have images only of dress or have multiple people in the same image. Such images need to be discarded or modified.

• Deep Fashion(48k)
• Categorization Dress Pattern (16k)
• Female Model images(2k)
• Apparel Dataset(16k)
• Bridal Dress(622)
• Dress Recognition(7k)
• Dress Fashion(60k)
• Dress Class Color(8k)
• Trousers(1.4k)
• Apparel Images Dataset(11k)
• Full-body-mads-dataset(1.2k)
• Full-body-tiktok-dancing-dataset(2.6k)
• somaset- synthetic humans(100k)
• Agender, full body image of people in the wild(4.6k)
• Everybody dance now, single person,full body and pose keypoints(83.9k)
• Everybody dance now(40.3k)
• Everybody dance now(35.4k
• Yoga Poses(1.5k)
• Yoga Poses - Large(6k)
• Human 3.6m

StyleGAN2 code is based on official StyleGAN2-ada rep