search

TrojanXu / yolov5-tensorrt

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

reshape error #13

Closed pkraison closed 4 years ago

pkraison commented 4 years ago

Traceback (most recent call last): File "main.py", line 212, in trt_result = profile_trt(build_engine(onnx_path, using_half), batch_size, 10, 100) File "main.py", line 160, in profile_trt return [np.array(yolo_outputs[0].host.reshape(1, -1, 85))] ValueError: cannot reshape array of size 123480 into shape (1,newaxis,85) Segmentation fault (core dumped)

TrojanXu commented 4 years ago

https://github.com/TrojanXu/yolov5-tensorrt/issues/10 related to this one. I will push a new commit to fix this later. For now, please change 85 to nc+5.

pkraison commented 4 years ago

thanks!