Runtime error while training model =1

knazeri / edge-connect

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

http://openaccess.thecvf.com/content_ICCVW_2019/html/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.html

Other

2.5k stars 530 forks source link

Runtime error while training model =1 #103

Open krazzzyKITTY opened 5 years ago

krazzzyKITTY commented 5 years ago

error

While running with our own dataset we are getting the following error

songzihang commented 4 years ago

Hello~ I have the same problem with you . I wonder if you solve it~

GuilmK commented 4 years ago

I find the solution. Because the input mask is 3 channels, it makes the other temp pictures become 3 channels. You can change the mask type to 1(random block) to solve this. Or change your mask channel to 1 by using opencv or other tools