YOLOFace로 얼굴 검출을 한다(Windows10, Python3.6)

소개



리얼타임의 물체 검출 알고리즘인 YOLOv3의 얼굴 검출판인 YOLOFace를 해 보았습니다. CPU에서도 움직일거야.

시스템 환경


  • Windows10(RTX2080 Max-Q, i7-8750H, RAM16GB)
  • Anaconda 2020.02
  • Python 3.6

  • 소개



    yoloface 을 복제합니다.

    yoloface 환경을 만듭니다.
    $ cd yoloface-master 
    $ conda create -n yoloface python=3.6
    $ conda activate yoloface
    $ pip install -r requirements.txt
    

    yolov3-wider_16000.weights을 다운로드하여 model-weights 폴더에 놓습니다.

    아래의 명령을 실행하여 얼굴 감지가 가능한지 확인합시다. samples 폴더에 얼굴 검출하고 싶은 파일을 넣어 둡니다.
  • 이미지의 경우
  • $ python yoloface.py --image samples/outside_000001.jpg --output-dir outputs/
    




    꽤 정확합니다!
  • 동영상의 경우
  • $ python yoloface.py --video samples/hoshino.mp4 --output-dir outputs/
    


  • 웹캠의 경우 (숫자를 변경하여 카메라 장치를 전환 할 수 있음)
  • $ python yoloface.py --src 1 --output-dir outputs/
    



    실시간으로 얼굴 검출이 가능했습니다.
    수고하셨습니다.

    좋은 웹페이지 즐겨찾기