Use YOLOv5 as a detector

stefanopini / simple-HRNet

Multi-person Human Pose Estimation with HRNet in Pytorch

GNU General Public License v3.0

573 stars 106 forks source link

Use YOLOv5 as a detector #106

Closed yan-ctrl closed 1 year ago

yan-ctrl commented 1 year ago

Hello, thank you for your excellent work. Can you tell me how to use YOLOv5 as a detector, or can you provide an explanation?

stefanopini commented 1 year ago

Hi @yan-ctrl , if you want to use YOLOv5, you can just set the parameters yolo_version='v5', yolo_model_def='yolov5n', as shown in this Google Colab notebook.

yan-ctrl commented 1 year ago

Okay, thank you