search

AlamiMejjati / Unsupervised-Attention-guided-Image-to-Image-Translation

Unsupervised Attention-Guided Image to Image Translation
MIT License
329 stars 47 forks source link

About attention model #6

Closed nile649 closed 6 years ago

nile649 commented 6 years ago

Hi, could you explain your attention model more in detail?

AlamiMejjati commented 6 years ago

Hi Nilesh, please refer to our paper https://arxiv.org/pdf/1806.02311.pdf for detailed explanations.

nile649 commented 6 years ago

Hi, I did read the paper, but I wish to just train the attention network. I tried to implement the attention network only, and it seems not giving better segmentation.

AlamiMejjati commented 6 years ago

Our attention network is an autoencoder formed by a series of convolutions followed by a residual block then a series of upsampling+convolutions are performed to obtain a continuous mask.