I am trying to retrain my own detection model on a Jetson Xavier NX, I have used camera-capture to grab some images and I can run train_ssd.py without any problems however when I then run onnx_export.py I get a string of errors like Traceback (most recent call last): File “onnx_export.py”, line 8…

Onnx_export.py outputs size mismatch for classification_headers.0.weight / bias errors

chbrain November 28, 2020, 1:50pm 2

Topic		Replies	Views
Python onnx_export.py shows error while trying to export model, please help Jetson Nano onnx	9	1236	January 11, 2022
Export onnx.py is failing for own camera captured data sets Jetson Nano onnx , nano2gb	5	1245	October 15, 2021
Evaluation of ssd mobile net Jetson Orin NX jetson-inference	2	19	February 12, 2025
Imagenet cannot operate with my exported in ONNX-model. May be I done something wrong/ Could you help? Computer Vision & Image Processing	0	336	October 24, 2022
ONNX model with Jetson-Inference using GPU Jetson Xavier NX tensorrt , jetson-inference , onnx	38	5708	October 18, 2021
Scores and boxes of onnx model exporting by jetson-inference Jetson Xavier NX jetson-inference	4	381	October 18, 2021
IsADirectoryError while running onnx_export.py Jetson Nano onnx	4	818	May 3, 2022
Using custom built model with ImageNet converted from h5 to onnx Jetson Nano	3	613	October 10, 2021
Trying to regenerate onnx for Jetson Nano Jetson Nano onnx	7	1722	October 18, 2021
Insufficient memory when exporting Yolo model from PyTorch to TensorRT Jetson Xavier Xavier NX Jetson Xavier NX tensorrt , yolo , pytorch	2	677	January 19, 2023