Different segmentation mask between Jetson Nano and Desktop GPU

[fredericolms]

The images from Jetson Nano look the same, but those two last images are not saved. The output on Jetson, follows:

And the console output:

batchDims.batchSize = 1, scale = 0.003922, format = 0
batchDims.batchSize = 1, scale = 0.003922, format = 0
batchDims.batchSize = 1, scale = 1.000000, format = 0
batchDims.batchSize = 1, scale = 1.000000, format = 0