RuntimeError: cannot reshape tensor of 0 elements into shape [1, 0, -1] because the unspecified dimension size -1 can be any value and is ambiguous about shift-net_pytorch HOT 3 OPEN

It seems the data is corrupted or the size of some images are no larger than fineSize, causing here a empty tensor.

from shift-net_pytorch.

Hi, thanks for the answer. I have solved the problem and it was indeed a problem with the mask data. I have another question, when I finished training the models for the three datasets: celeba, PSV, and place, how should I select the corresponding model I trained for the test?

from shift-net_pytorch.

If the test image is a face, using celeba model, I do not know the PSV dataset. If the test image is about the natural scenery, using the place model. Here’s a piece of advice for you: maybe you need to try more advanced inpainting methods. This repo is old, you can use it for learning, not suitable for production or practical use.

from shift-net_pytorch.