fixed contextual attention module (various problems & discrepancies with baseline implementation)

WonwoongCho / Generative-Inpainting-pytorch

This is a pytorch version of the Generative Image Inpainting with Contextual Attention

71 stars 14 forks source link

Open ShnitzelKiller opened 5 years ago

ShnitzelKiller commented 5 years ago

Fixed extract_patches() to permute the dimensions so that the subsequent views are correct (Issue #2)
Fixed mask patch processing, which was inconsistent with image patches
Fixed normalization of convolution weights
Fixed downsampling with non-unity rate
Added runnable test in model_module.py to test the attention module