Vious / LBAM_Pytorch

Pytorch re-implementation of Paper: Image Inpainting with Learnable Bidirectional Attention Maps (ICCV 2019)

Home Page:http://openaccess.thecvf.com/content_ICCV_2019/papers/Xie_Image_Inpainting_With_Learnable_Bidirectional_Attention_Maps_ICCV_2019_paper.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About the training mask set

haolin512900 opened this issue · comments

Hi ! Thank you for your great project!
Can you tell me the detaill of training mask set? What about the 18,000 masks you mentioned in your paper?

Hi, thanks for your interest. The 12,000 irregular masks are from https://nv-adlr.github.io/publication/partialconv-inpainting.
I have found the previous script for generating masks, and uploaded it at https://github.com/Vious/LBAM_Pytorch/blob/master/generateMask.py .
I generate 6,000 masks using this script, you could modify the code to get different mask portions like 25%-50%.

Hi, thanks for your interest. The 12,000 irregular masks are from https://nv-adlr.github.io/publication/partialconv-inpainting.
I have found the previous script for generating masks, and uploaded it at https://github.com/Vious/LBAM_Pytorch/blob/master/generateMask.py .
I generate 6,000 masks using this script, you could modify the code to get different mask portions like 25%-50%.

Thank your reply!Good lucky!