search

tsing90 / pytorch_semantic_human_matting

This is an unofficial implementation of the paper "Semantic human matting":
https://arxiv.org/pdf/1809.01354.pdf
83 stars 18 forks source link

test M net #17

Closed Oseira closed 5 years ago

Oseira commented 5 years ago

Hi tsing, I am trying to use the test.py file. There is no problem with T_net but when I try with M-net I get an error: "indices and input shapes do not match: indices [1 x 64 x 225 x 400], input [1 x 64 x 224 x 400]" when trying to do the last pooling layer of the decoder. Do you know why I am getting this 2 pixels difference?

Oseira commented 5 years ago

it has to be a multiple of 64

sanshibayuan commented 5 years ago

I change the size in test.py to 512 and I still got the eroor