marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.39k stars 344 forks source link

segmentation fault (core dumped) #430

Open ghty11468 opened 10 months ago

ghty11468 commented 10 months ago

587ff9fbef59b4bb4c98760bceab29b

The engine file has been generated, but this error always occurs during inference. What should I do?

marcoslucianops commented 10 months ago

Please follow: https://github.com/marcoslucianops/DeepStream-Yolo/blob/master/docs/YOLOv5.md

The engine output is wrong, so you are using the wrong ONNX model.

ghty11468 commented 10 months ago

请关注: https://github.com/marcoslucianops/DeepStream-Yolo/blob/master/docs/YOLOv5.md

引擎输出错误,因此您使用了错误的 ONNX 模型。

My onnx file was generated using the Ultratics/Yolov5 project, but is it not allowed?

marcoslucianops commented 10 months ago

You need to use the export_yoloV5.py file from the utils folder in this repo to export the ONNX model from the YOLOv5 PyTorch model (as said in the link I sent you, please follow it).

emad555 commented 10 months ago

You need to use the export_yoloV5.py file from the utils folder in this repo to export the ONNX model from the YOLOv5 PyTorch model (as said in the link I sent you, please follow it).

This does not generate an engine file, only onnx file.

marcoslucianops commented 10 months ago

The engine file is generated in the first time you run the deepstream-app command.

emad555 commented 10 months ago

Thank you, it is working now!