yolov8_pose_trt.py

wang-xinyu · Apr 29, 2024 · c6110dd · c6110dd
1 parent 1744711
commit c6110dd
Show file tree

Hide file tree

Showing 2 changed files with 504 additions and 3 deletions.
diff --git a/yolov8/README.md b/yolov8/README.md
@@ -129,9 +129,10 @@ sudo ./yolov8_pose -d yolov8n-pose.engine ../images g //gpu postprocess
 ```
 // install python-tensorrt, pycuda, etc.
 // ensure the yolov8n.engine and libmyplugins.so have been built
-python yolov8_det.py  # Detection
-python yolov8_seg.py  # Segmentation
-python yolov8_cls.py  # Classification
+python yolov8_det_trt.py  # Detection
+python yolov8_seg_trt.py  # Segmentation
+python yolov8_cls_trt.py  # Classification
+python yolov8_pose_trt.py  # Pose Estimation
 ```
 
 # INT8 Quantization