search

IS2AI / TFW

TFW: Annotated Thermal Faces in the Wild Dataset
MIT License
22 stars 3 forks source link

Missing file detect_face.py #3

Closed TadeasFrycak closed 1 year ago

TadeasFrycak commented 1 year ago

Hello,

I would really like to use your Face models, but I cannot find the detect_face.py file mentioned in README, would you please share it with me?

Thank you for your work!

akuzdeuov commented 1 year ago

Hello,

Thank you for your interest in our project. The detect_face.py file is located in the original YOLO5Face repo: First, you need to download YOLOv5 repo and install the required packages: 

$ git clone https://github.com/ultralytics/yolov5 
$ cd yolov5
$ pip install -r requirements.txt

Then, download the YOLO5Face repo:

$ git clone https://github.com/deepcam-cn/yolov5-face.git
$ cd yolov5-face

To use our pre-trained face models, download this file from Google Drive and unzip it in the yolo5-face/weights directory. Then, run the following script:

$ python detect_face.py --weights PATH_TO_MODEL --source PATH_TO_IMAGE
TadeasFrycak commented 1 year ago

It works grandly, thank you very much!