knazeri / edge-connect

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

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Sizes of tensors must match except in dimension 1

tendrillion opened this issue · comments

When testing your own pictures with the shape [3,577,900],this code error which in line 251 in the src/models.py :

the error is :Sizes of tensors must match except in dimension 1. Expected size 557 but got size 556 for tensor number 1 in the list.

the code line 140 in src/models.py:
outputs = self.generator(inputs)
the shape of inputs is :[1,3,557,900]
but the shape of outputs is :[1,1,556,900]

any suggestions to solve this?