xiezhy6 / PASTA-GAN-plusplus

34 stars 5 forks source link

Running the code #1

Open generalbigsur opened 2 years ago

generalbigsur commented 2 years ago

Hello, Great work! The results in the paper look very impressive. Could you please provide instructions on how to run this code? Also, could you provide the links to images in the dataset and instructions to do any preprocessing? Thank you very much!

xiezhy6 commented 2 years ago

Sure, we will add a readme about the instruction this weekend :-)

niranjan-v commented 2 years ago

Hi, Thanks a lot for providing the code. Could you kindly confirm which package has been used to generate the garment parsing inputs in the test_datas? This would help to train and test the method on custom data. Thanks!

generalbigsur commented 2 years ago

+1. Yeah, a section on "data preparation" similar to one in the PASTA-GAN repo would be really helpful. Please consider adding that too. Thank you very much!

xiezhy6 commented 2 years ago

Hi, Thanks a lot for providing the code. Could you kindly confirm which package has been used to generate the garment parsing inputs in the test_datas? This would help to train and test the method on custom data. Thanks!

Hi, like our previous work PASTA-GAN, the sleeve mask is not a necessary input for the model training and testing. I will update the codebase to make the garment parsing an optional input for the model training and testing. :-)

buptpriswang commented 2 years ago

Thanks a lot for providing the code! But I got the following error when I run the code. Loading networks from "./ckpt/network-snapshot-004408.pkl"... Loading custom kernel... Traceback (most recent call last): File "test.py", line 190, in generate_images() # pylint: disable=no-value-for-parameter File "/home/qinlong.wang/anaconda3/envs/pasta37/lib/python3.7/site-packages/click/core.py", line 1130, in call return self.main(args, kwargs) File "/home/qinlong.wang/anaconda3/envs/pasta37/lib/python3.7/site-packages/click/core.py", line 1055, in main rv = self.invoke(ctx) File "/home/qinlong.wang/anaconda3/envs/pasta37/lib/python3.7/site-packages/click/core.py", line 1404, in invoke return ctx.invoke(self.callback, ctx.params) File "/home/qinlong.wang/anaconda3/envs/pasta37/lib/python3.7/site-packages/click/core.py", line 760, in invoke return __callback(args, *kwargs) File "/home/qinlong.wang/anaconda3/envs/pasta37/lib/python3.7/site-packages/click/decorators.py", line 26, in new_func return f(get_current_context(), args, **kwargs) File "test.py", line 103, in generate_images G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore File "/home/qinlong.wang/exp/virtual_try_on/PASTA-GAN-plusplus/legacy.py", line 21, in load_network_pkl data = _LegacyUnpickler(f).load() File "/home/qinlong.wang/exp/virtual_try_on/PASTA-GAN-plusplus/torch_utils/persistence.py", line 191, in _reconstruct_persistent_obj module = _src_to_module(meta.module_src) File "/home/qinlong.wang/exp/virtual_try_on/PASTA-GAN-plusplus/torch_utils/persistence.py", line 227, in _src_to_module exec(src, module.dict) # pylint: disable=exec-used File "", line 2282, in NameError: name 'os' is not defined

buptpriswang commented 2 years ago

Hi, Thanks a lot for providing the code. Could you kindly confirm which package has been used to generate the garment parsing inputs in the test_datas? This would help to train and test the method on custom data. Thanks!

Hi, like our previous work PASTA-GAN, the sleeve mask is not a necessary input for the model training and testing. I will update the codebase to make the garment parsing an optional input for the model training and testing. :-)

hello, is there any update about this?

tykji93 commented 2 years ago

hello. is there any update?

xiezhy6 commented 2 years ago

hello. is there any update?

Hi, we have updated the test scripts. Now you can run the test code without the parameter '--use-sleeve-mask' if the sleeve mask is unavailable in your case.