binnie723 / CJ-AI-Vision-PBL

yolov5를 사용해서 주인공을 제외한 사람들 얼굴에 당근 이미지 대체 프로젝트

Head & Person Detection Model

Model Github Link: https://github.com/deepakcrk/yolov5-crowdhuman
Model Download Link: YOLOv5m-crowd-human

Test

$ python detect.py --weights crowdhuman_yolov5m.pt --source _test/ --view-img  --heads

Output