PRBonn / bonnet

Bonnet: An Open-Source Training and Deployment Framework for Semantic Segmentation in Robotics.
GNU General Public License v3.0
325 stars 89 forks source link

CWC pre-trained model return wrong prediction on dataset image #56

Open Guillaumebeuzeboc opened 5 years ago

Guillaumebeuzeboc commented 5 years ago

Hello, First, thank you for the awesome work! I am trying to run the pre-trained model of the CWC in the docker. I downloaded the pre-trained model and froze it. So I tried the inference on an image from the dataset (I tried different images). And the result is not the one expected: 001_image_mask

It seems that most of the background is detected as weed and that no crops are detected. I was expecting something close to the mask from the dataset. I tried with tf and trt as backend and both return the same output. From the readme we can see that the output should be way better. Any idea why I am getting this output? Thanks,

Guillaume