search

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
Apache License 2.0
190 stars 46 forks source link

cannot convert,help!!! #4

Closed yushanshan05 closed 4 years ago

yushanshan05 commented 4 years ago

I use pytorch（1.4.0），onnx（1.6.0），tensorrt（6.0）. I can conver onnx form torch sucessful. But cant conver onnx to tensorrt.

These are the erros: [TensorRT] ERROR: (Unnamed Layer* 0) [Slice]: slice is out of input range ERROR: Failed to parse the ONNX file. In node 9 (convert_axis): UNSUPPORTED_NODE: Assertion failed: axis >= 0 && axis < nbDims Traceback (most recent call last): File "main.py", line 180, in profile_trt(buildengine('yolov5{}.onnx'.format(batch_size), using_half), batch_size) File "main.py", line 94, in profile_trt assert(engine is not None) AssertionError

TrojanXu commented 4 years ago

Sorry I gave the wrong message in the initial commit. Please use pytorch1.4.0+onnx1.6.0+trt7.0 to see if this works.