본문 바로가기

Data-science/deep learning

동물(개, 강아지) 얼굴만 추출, pretrained detection model 이용하기

728x90

github.com/kairess/dog_face_detector

 

kairess/dog_face_detector

Detect dog face rect and facial landmarks(6 points) using dlib - kairess/dog_face_detector

github.com

detector만 필요하다. cnn base의 얼굴 검출기다.

여러 개가있는 이미지를 입력하고.

Detector에서 검출된 Rectangle에 빨간 테투리를 그려준다.

그려준 후, (내가 추가한 코드) 원하는 크기만큼 잘라서 얼굴 이미지를 원하는 사이즈로 리사이징한 후 저장해준다.

빨간 사각형이 잘 그려졌다.

강아지 얼굴만 저장하기! 그럭저럭 잘 된듯하다. 학습 하나 안 하고도 이렇게 검출이 가능합니다.