search

elyha7 / yoloface

Yolov5 Face Detection
GNU General Public License v3.0
123 stars 36 forks source link

Where can I find weight for different models? yolov5n-face, yolov5s-face, yolov5m-face and so on? #12

Open glorsh66 opened 1 year ago

glorsh66 commented 1 year ago

Where can I find weight for different models? yolov5n-face, yolov5s-face, yolov5m-face and so on?

elyha7 commented 1 year ago

You can download model weights from deepcam-cn original repo: https://github.com/deepcam-cn/yolov5-face#pretrained-models. Follow this section of readme: https://github.com/elyha7/yoloface#other-pretrained-models. Don't forget to save weight as pytorch state_dict.