• Hardware Platform (Jetson / GPU): Jetson Xavier AGX
• DeepStream Version: 5.1
• JetPack Version (valid for Jetson only): 4.5.1
• TensorRT Version: 7.1.3
Hello , I am trying to run deepstream-segmentation python code on AGX.
Here is the GitHub link: (deepstream_python_apps/apps/deepstream-segmentation at master · NVIDIA-AI-IOT/deepstream_python_apps · GitHub)
I put the trained model I trained, also edit the “dstest_segmentation_config_semantic.txt”
But I met a problem, below is the problem I met, how can I solve it, thanks.
I find something strange in the error message.
The thing is that the model I exported its output node is “Softmax_1 (Softmax)” and its output shape is “(None, 720, 1280, 19)”, show below.
The code in “deepstream_segmentation.py” is “bgr = np.zeros((shp[0], shp[1], 3))”, I print the shp[0] and shp[1] and get 1280 and 19, I think the correct ones should be gotten are 720 and 1280.
The output node after training is “conv2d_5 (Conv2D)” and shape is"(19, 720, 1280)".
But when I export the model will get two more nodes named “permute_1 (Permute)” with shape(720, 1280, 19) and “Softmax_1 (Softmax)” with shape(720, 1280, 19).
The output node shape just change the order from (19,720, 1280) to (720, 1280, 19), thus when I do inferences on AGX it will get (1280, 19) instead (720, 1280), and the error happened.
Why it change the order?
Is there anything can do to maintain the output node shape’s order (19,720,1280)?