Please provide the following info (check/uncheck the boxes after creating this topic):
Software Version
DRIVE OS Linux 5.2.6
DRIVE OS Linux 5.2.6 and DriveWorks 4.0
DRIVE OS Linux 5.2.0
DRIVE OS Linux 5.2.0 and DriveWorks 3.5
DRIVE OS Linux 5.1.6.1 and DriveWorks 2.2
NVIDIA DRIVE™ Software 10.0 (Linux)
NVIDIA DRIVE™ Software 9.0 (Linux)
other DRIVE OS version
other
Target Operating System
Linux
QNX
other
Hardware Platform
NVIDIA DRIVE™ AGX Xavier DevKit (E3550)
NVIDIA DRIVE™ AGX Pegasus DevKit (E3550)
other
SDK Manager Version
1.8.0.10363
other
Host Machine Version
native Ubuntu 18.04
other
Hi,
I have a problem with implementation custom model dnn (yolo v4 or yolov3) for example: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
I try convert yolo model (cfg, weights) to onnx model - GitHub - Tianxiaomo/pytorch-YOLOv4: PyTorch ,ONNX and TensorRT implementation of YOLOv4
I installed onnx=1.6.0 and torch=1.2.0 or 1.3.0 or 1.10.2 and I can convert to onnx model
python3 demo_darknet2onnx.py cfg/yolov4.cfg data/coco.names yolov4.weights data/dog.jpg -1
but I have a problem with convert onnx model to tensorrt model using tensorRT_optimization
/usr/local/driveworks/tools/dnn/tensorRT_optimization --modelType=onnx --onnxFile=yolov4.onnx --out=yolov4.bin
I have a lot of problems, for example:
Initializing network optimizer on model yolov4.onnx.
----------------------------------------------------------------
Input filename: yolov4.onnx
ONNX IR version: 0.0.7
Opset version: 10
Producer name: darknet to ONNX example
Producer version:
Domain:
Model version: 0
Doc string:
----------------------------------------------------------------
WARNING: ONNX model has a newer ir_version (0.0.7) than this parser was built against (0.0.3).
While parsing node number 201 [Resize]:
ERROR: ModelImporter.cpp:147 In function importNode:
[8] No importer registered for op: Resize
Error: DW_FILE_INVALID: Unable to parse given onnx file.
or
Initializing network optimizer on model yolov4.onnx.
----------------------------------------------------------------
Input filename: yolov4.onnx
ONNX IR version: 0.0.7
Opset version: 9
Producer name: pytorch
Producer version: 1.10
Domain:
Model version: 0
Doc string:
----------------------------------------------------------------
WARNING: ONNX model has a newer ir_version (0.0.7) than this parser was built against (0.0.3).
While parsing node number 0 [Conv]:
ERROR: ModelImporter.cpp:288 In function importModel:
[5] Assertion failed: tensors.count(input_name)
Error: DW_FILE_INVALID: Unable to parse given onnx file.
or
Initializing network optimizer on model yolov3.onnx.
----------------------------------------------------------------
Input filename: yolov3.onnx
ONNX IR version: 0.0.4
Opset version: 9
Producer name: pytorch
Producer version: 1.2
Domain:
Model version: 0
Doc string:
----------------------------------------------------------------
WARNING: ONNX model has a newer ir_version (0.0.4) than this parser was built against (0.0.3).
WARNING: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
Successfully casted down to INT32.
While parsing node number 215 [Gather]:
ERROR: onnx2trt_utils.hpp:277 In function convert_axis:
[8] Assertion failed: axis >= 0 && axis < nbDims
Error: DW_FILE_INVALID: Unable to parse given onnx file.
or
Initializing network optimizer on model yolov3.onnx.
----------------------------------------------------------------
Input filename: yolov3.onnx
ONNX IR version: 0.0.4
Opset version: 8
Producer name: pytorch
Producer version: 1.2
Domain:
Model version: 0
Doc string:
----------------------------------------------------------------
WARNING: ONNX model has a newer ir_version (0.0.4) than this parser was built against (0.0.3).
While parsing node number 214 [Cast]:
ERROR: builtin_op_importers.cpp:704 In function importCast:
[8] Assertion failed: inputs.at(0).is_tensor()
Error: DW_FILE_INVALID: Unable to parse given onnx file.
Can you help what I doing wrong?
I tried do this simple tutorial: GitHub - ShayNvidia/dnnWebinar: NVIDIA DriveWorks for Drive
and I have a bin model and it working fine.