I trained a mobilenet-based segmentation model using google colab then deployed it on the Jetson TX2. Then I run a test video using the model on the TX2 and a laptop (GTX 960M). Unfortunately the result I got from the TX2 is much worse than from the laptop. I tried switching off GPU and run the model using CPU only but the result is still the same
This is my laptop’s result, which it’s supposed to be:
And this is the TX2’s result:
Here is my mobilenet architecture:
My TX2 configuration: Ubuntu 16.04, Jetpack 3.3, Tensorflow-gpu 1.10, CUDA 9.0, cudnn 7.1.5
Please help me to solve this problem! Thank you!