Inference with custom dataset??

SHShim0513 / SD-VITON

[AAAI 2024] Towards Squeezing-Averse Virtual Try-On via Sequential Deformation.

https://arxiv.org/pdf/2312.15861.pdf

115 stars 14 forks source link

Inference with custom dataset?? #5

Open lynguyenminh opened 5 months ago

lynguyenminh commented 5 months ago

Thank you for your great work!

Does this source code support inferencing on custom data?

syedhaideralizaidi commented 4 months ago

Can you please guide me how to run this? I downloaded model weights and then run the command "python3 train_generator.py --name test -b 4 -j 8 --gpu_ids 0,1 --fp16 --tocg_checkpoint ./checkpoints/test/tocg.pth --occlusion --fine_width 384 --fine_height 512 --num_upsampling_layers more --cond_G_ngf 48 --cond_G_input_width 384 --cond_G_input_height 512 --cond_G_num_layers 6". It's not working.