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

Batch shift #88

Closed Zhaoyi-Yan closed 5 years ago

Zhaoyi-Yan commented 5 years ago

It gives the same results of the origin code(both in forward and backward) and is faster when batchsize is large(0.049s/image --> 0.025s/image when batchsize=64)