search

marcoslucianops / DeepStream-Yolo-Pose

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Pose models
MIT License
109 stars 25 forks source link

The engine can only predict no any class #11

Closed JosephKL-Chan closed 7 months ago

JosephKL-Chan commented 7 months ago

Our organization had the issue that we use Yolov8n model, we use deepstream-yolo change it to onnx format. and we follow the https://github.com/marcoslucianops/DeepStream-Yolo-Pose/blob/master/docs/YOLOv8_Pose.md YOLOv8-Pose usage step to setup the deepstream. But only one class can detected. Our process was:

  1. python3 export_yoloV8_pose.py -w yolov8s-pose.pt --dynamic --simplify
  2. Copy onnx to deepstream folder.
  3. Use onnx change engine ./trtexec --onnx=yolov8npose.onnx --saveEngine=yolov8npose.engine
  4. Compile nvdsinfer_custom_impl_Yolo plugin
  5. Edit the config_infer_primary_yoloV8_pose.txt (... onnx-file=yolov8npose.onnx model-engine-file=yolov8npose.engine network-mode=0 num-detected-classes=3 network-type=3 cluster-mode=4 ... parse-bbox-func-name=NvDsInferParseYoloPose) ... [class-attrs-all] pre-cluster-threshold=0.25 topk=300
  6. Edit label.txt
  7. Run RTSP Programe