WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.02k stars 4.12k forks source link

How to create a single model for person, face and pose detection. #888

Open pk-1196 opened 1 year ago

pk-1196 commented 1 year ago

In the yolov7 we have good detection of person , face and pose of the person separately . Is there any way to do that using single model or in one go altogether

Owaiskhan9654 commented 1 year ago

Hey @pk-1196, Check out this Kaggle Notebook for custom dataset as you want- Note - Here you will be needing your custom dataset labeled on Roboflow if you directly run this Notebook in Kaggle environment.

https://www.kaggle.com/code/owaiskhan9654/training-yolov7-on-kaggle-on-custom-dataset

Do comment if you still face issue. Happy to help 😊

Thanks Owais

pk-1196 commented 1 year ago

@Owaiskhan9654 the kaggle notebook is only for object detection , i am asking for person detection and its pose keypoint detection together

Owaiskhan9654 commented 1 year ago

Like this you can do and choose whichever points you want to label in pose detection. Using single model yolov7-w6-pose @pk-1196 image