Zhaoyi-Yan / Shift-Net_pytorch

Pytorch implementation of Shift-Net: Image Inpainting via Deep Feature Rearrangement (ECCV, 2018)
http://openaccess.thecvf.com/content_ECCV_2018/papers/Zhaoyi_Yan_Shift-Net_Image_Inpainting_ECCV_2018_paper.pdf
MIT License
363 stars 83 forks source link

pretrain model? #7

Open junhocho opened 6 years ago

junhocho commented 6 years ago

It would be helpful if you provide pretained model. Thank you.

Zhaoyi-Yan commented 6 years ago

Hi, our original code is base on Torch7, https://github.com/Zhaoyi-Yan/Shift-Net . We haven't got that performance in pytorch. But we promise, there is no much gap between these two versions.

amjltc295 commented 5 years ago

May I get the pretrained Pytorch model even if it may not be as good as the Torch7 one? Thanks.