Please provide complete information as applicable to your setup.
• Hardware Platform (Jetson / GPU) V100-SXM2 • DeepStream Version • JetPack Version (valid for Jetson only) • TensorRT Version 7.0.0 • NVIDIA GPU Driver Version (valid for GPU only) 510.47.03 • Issue Type( questions, new requirements, bugs) • How to reproduce the issue ? (This is for bugs. Including which sample app is using, the configuration files content, the command line used and other details for reproducing)
I am trying to convert http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v2_coco_2018_03_29.tar.gz to tensorRT. Following the path tensorflow → onnx → tensorRT
I ran this command
./trtexec --onnx=/home/sahilwa/workplace/onnx-models/ssd_mobilenet_v2_coco_2018_03_115.onnx --explicitBatch --workspace=4096 --minTiming=10 --fp16 --saveEngine=ssd_mobilenet_v2_coco_2018_03_115.trt --verbose
But I get this error
[8] Assertion failed: convertOnnxWeights(initializer, &weights, ctx)
• Requirement details( This is for new requirement. Including the module name-for which plugin or for which sample application, the function description)
Hi,
Request you to share the ONNX model and the script if not shared already so that we can assist you better.
Alongside you can try few things:
validating your model with the below snippet
check_model.py
import sys
import onnx
filename = yourONNXmodel
model = onnx.load(filename)
onnx.checker.check_model(model).
2) Try running your model with trtexec command.
In case you are still facing issue, request you to share the trtexec “”–verbose"" log for further debugging
Thanks!