Zhangjinso / PISE

123 stars 28 forks source link

Do you have a trained model? #29

Closed freedom521jin closed 3 years ago

freedom521jin commented 3 years ago

I want to only test and i don't have any gpus,so,please provide a model that can be used to test? Thx a lot!

Zhangjinso commented 3 years ago

I think you can modified the set input function to change the inputs from 'cuda' to 'cpu' and test as before to get the results (refer to this. ).

Hope this can help you~

freedom521jin commented 3 years ago

I think you can modified the set input function to change the inputs from 'cuda' to 'cpu' and test as before to get the results (refer to this. ).

Hope this can help you~

Thx, I can test with cpu.And , I have trained the model ,follew your steps ,but some evel results are not very satisfactory ,the model has been trained for one week by GPU-1080Ti.Then, I download your evel results,and some results that I test by my model are bad too . So I hope you can provide your model that trained already.Thx!

Zhangjinso commented 3 years ago

The checkpoint used in our paper was released in ReadMe.

Pre-trained checkpoint of human pose transfer reported in our paper can be found from baidu (fetch code: abcd) or Google drive and put it in the folder (-->results-->fashion).