dressing-in-order icon indicating copy to clipboard operation
dressing-in-order copied to clipboard

Test for high resoluton images

Open kieutrongthien opened this issue 3 years ago โ€ข 11 comments

I have two images with size 768x1024. How i can test try on but still keep high resolution ?. Do i need to train model with 768x1024 before test it ?.

kieutrongthien avatar Feb 07 '22 09:02 kieutrongthien

The released model is only train and test on 256x176, so for higher resolution, it may not have the same behavior and you may need to re-train and tune the hyperparameters.

cuiaiyu avatar Feb 07 '22 16:02 cuiaiyu

I think i can try to increase image quality after i have result image from try on. Because my server only have 4 GPU Tesla T4 16GB, it take to 3 - 4 day to train on current size 256 x 176 and use 10GB Memory on GPU 0 and 3GB to the every GPU left. I tried to change size of images to 512 x 384 but it show error CUDA memory not enough.

Thanks for your answer, it's give me another solution for this case.

kieutrongthien avatar Feb 15 '22 18:02 kieutrongthien

Thank you for the model provided 256 x 176, Can you release a model trained on 512x352 ?

tantom avatar Mar 07 '22 06:03 tantom

Ah, I suspect training on a single GPU for 256x176 would be faster than 4 GPUs.

Unfortunately the code is not well optimized for parallel training, so the memory is not well balanced. If you want to do some optimization, you may try to wrap all separate models (like E_attr, Flow, G) into a unified model before calling nn.Parallel and get rid of those function calls like model.netG.module.something.

cuiaiyu avatar Mar 09 '22 21:03 cuiaiyu

Hi, is it possible to resize the image into high resolution, like 512*352, and put it into the pretrained models? Thank you

NorthanX avatar Jul 05 '22 06:07 NorthanX

@NorthanX You need to re-train model in higher resolution to make it work

kieutrongthien avatar Jul 05 '22 06:07 kieutrongthien

@kieutrongthien Thanks! So have you done your high quality training? I am still trying to change the parameters of the training process and I am quite curious about the results.

NorthanX avatar Jul 05 '22 07:07 NorthanX

@NorthanX you can use dataset provided in README:

http://mmlab.ie.cuhk.edu.hk/projects/DeepFashion/InShopRetrieval.html

I trained on higher resolution before but it have a little problem at the face of human in the result, you can check here: #32

So if you need to improve it, you need to prepare your own dataset with more difference human models.

kieutrongthien avatar Jul 05 '22 07:07 kieutrongthien

@kieutrongthien I will train with the original deepfashion inshop dataset first. Thanks for your advice!

NorthanX avatar Jul 05 '22 07:07 NorthanX

  1. Does this project work for you guys - when I tried it out some time ago I didnโ€™t manage to get the code running (canโ€™t remember the exact reason now)?

  2. It would be great if you could create a new repo here in Git with all the fixes and changes you did to make it work - especially with a higher resolution model included ๐Ÿค“๐Ÿ‘๐Ÿผ

tir. 5. jul. 2022 kl. 11.23 skrev NorthanX @.***>:

@kieutrongthien https://github.com/kieutrongthien I will train with the original deepfashion inshop dataset first. Thanks for your advice!

โ€” Reply to this email directly, view it on GitHub https://github.com/cuiaiyu/dressing-in-order/issues/33#issuecomment-1174710318, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA5IZ3PNIPONBFIYKATIV2DVSPPIVANCNFSM5NW5BSWQ . You are receiving this because you are subscribed to this thread.Message ID: @.***>

Garfield2013 avatar Jul 05 '22 07:07 Garfield2013

@Garfield2013

  1. The demo code works for me.
  2. I am not sure about the process of higher resolution training, and I am still trying to change the parameter. But if there is any progress, I will update a repo and let you know.

NorthanX avatar Jul 06 '22 02:07 NorthanX

Great! Thank you so much! ๐Ÿค“๐Ÿ‘๐Ÿผ

ons. 6. jul. 2022 kl. 06.24 skrev NorthanX @.***>:

@Garfield2013 https://github.com/Garfield2013

  1. The demo code works for me.
  2. I am not sure about the process of higher resolution training, and I am still trying to change the parameter. But if there is any progress, I will update a repo and let you know.

โ€” Reply to this email directly, view it on GitHub https://github.com/cuiaiyu/dressing-in-order/issues/33#issuecomment-1175706370, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA5IZ3PC2RIX2L2CMCIDOKTVSTU55ANCNFSM5NW5BSWQ . You are receiving this because you were mentioned.Message ID: @.***>

Garfield2013 avatar Oct 11 '22 08:10 Garfield2013

Hi @kieutrongthien I hope you are doing good. Have you trained on higher resolution images? I am looking for virtual tryon on high resolution images. Did you get same performance as 256x176 dataset?

I trained on higher resolution before but it have a little problem at the face of human in the result, you can check here: #32

vinodbukya6 avatar Oct 27 '22 12:10 vinodbukya6

Hi @kieutrongthien I hope you are doing good. Have you trained on higher resolution images? I am looking for virtual tryon on high resolution images. Did you get same performance as 256x176 dataset?

I trained on higher resolution before but it have a little problem at the face of human in the result, you can check here: #32

You can get same performance with higher resolution. I tried on 512x384 before, only 1 problem is the face of the result.

If you want to get better result, you need to put more images with difference faces in to the dataset.

kieutrongthien avatar Oct 27 '22 12:10 kieutrongthien

@kieutrongthien Thank you for quick response. If possible can you please share pre-trained models of high resolution images?

vinodbukya6 avatar Oct 27 '22 13:10 vinodbukya6

Hi @kieutrongthien Is it possible to share the models?

vinodbukya6 avatar Oct 28 '22 14:10 vinodbukya6

Hi @kieutrongthien Is it possible to share the models?

Sorry, I can't share pretrained model to you, my model trained on edited source code of this repository. So it will not work with original source code.

kieutrongthien avatar Nov 03 '22 06:11 kieutrongthien

Okay. No problems!!!

vinodbukya6 avatar Nov 03 '22 06:11 vinodbukya6