Intel Openvino Pretrained Model Unet-camvid-onnx-0001 Does Not Predict Correctly

I downloaded the Intel pre-trained Unet model from the OpenVINO Model Zoo Github repo without any modifications. But it does not seem working, please have a look the below predict

Solution 1:

The code seems fine. The model output of unet-camvid-onnx-0001 is the per-pixel probabilities of each input pixel of the 12 classes of the CamVid dataset. The RGB value of the 12 classes can be found in this directory: INSTALL_DIR\deployment_tools\open_model_zoo\data\palettes

You can refer to the classes for the RGB value in your code.