TidamCo

You probably have a greyscale image as an input, and the model is expecting RGB. (RGB has a shape of 3, while greyscale has a shape of 1)

I had the same problem so I checked in the code, the problem comes from the fact that your attributes are set in private, if you set them in protected...

I just ran into the same problem I think. If you read the code, you'll see this : `RESTORE_FROM = './checkpoint/CIHP_pgn'` Which mean you have to unzip the pre-trained model...