Mask_YOLO

2020F-IntroductionToDeepLearning-Term Project

---

Reference

1. YOLO Fine Tuning, https://medium.com/analytics-vidhya/covid-19-face-mask-detection-using-yolov5-8687e5942c81

2. YOLO dataset, https://github.com/iAmEthanMai/mask-detection-dataset.git

3. YOLOv5 Code, https://github.com/ultralytics/yolov5

---

Execution

Basic Exmaple

저희가 fine tuning한 weight는 weights 폴더 안에 best.pt로 저장되어 있습니다.

웹캠을 통해 detection을 실행하실려면 source argument를 0으로, 다른 사진이나 동영상에 detection을 실행하려면 DATA_PATH를 입력하시면 됩니다.

python detect.py --weights ./weights/best.pt --conf 0.4 --source 0

python detect.py --weights ./weights/best.pt --conf 0.4 --source DATA_PATH

---

NOTICE

requirements.txt 에서 torchvision>=0.8.1을 요구하고 있는데, conda 환경에서 pip로 설치가 안되는 오류를 확인해서 whl 파일을 첨부하였습니다.

현재 디렉토리에서 다음 코드를 통해 설치하실 수 있습니다.

python -m pip install whl_FILE_PATH

---

TEST RESULT

테스트 사진과 영상에 대한 detection 결과는 ./test_result 에서 확인하실 수 있습니다.

---

colab_codes

Fine Tuning에 사용된 코드를 확인하실 수 있습니다.