The parsing network does not give desired output about generativefacecompletion

[liaohaofu]

The issue is related to preprocessing. It looks like that, somehow for pycaffe, the dimensions of the input data should be totally reversed. I have to change

input_ = input_.transpose(2, 0, 1)

to

input_ = input_.transpose(2, 1, 0)

to make the code work. Thanks again for your help! @Yijunmaverick

[Yijunmaverick]

Hi @liaohaofu, thanks for your interests about our work.

I just run my model and the result (below) looks good. Not sure if it is due to the processing steps on input and ouput.

My guess is on this line "segmentation = scores.argmax(0)". Do you get the max along the channel (channel = 11) dimension?

[liaohaofu]

Yes, the max is along the channel dimension. In pycaffe, the channel dimension always comes first. So, "input_" has a shape of 1x3x128x128 (1 is the batch size) and "scores" has a shape of 11x128x128. Could you run my code on your machine to see if it works correctly. You only need to change the related file paths to make it work if you have your caffe compiled with anaconda. Thanks a lot!