Hi there,
Could the bad results be because you’re reconstructing the raw float tensor. Python represents the pixel values in [0, 1] while libraries like Caffe accept it in [0,255]. If that’s the case maybe multiplying the pixel values by a factor may help?
It’s a hunch because PSPNet, FCN and Segnet have all been built with Caffe.