DeepFace 모듈 소개

소개

안녕하세요! 이 기사에서는 deepface 모듈을 소개합니다.

DeepFace 모듈이란 무엇입니까?

DeepFace는 Python용 경량 얼굴 인식 및 얼굴 속성 분석(연령, 성별, 감정 및 인종) 프레임워크입니다.

간단히 말해서 deepface는 자체 모델 등을 교육할 필요 없이 다양한 기능을 분석할 수 있습니다.

환경 준비

먼저 이 예제에서 사용할 가상 환경을 초기화해야 합니다. 이 작업은 다음을 통해 수행할 수 있습니다.

python3 -m venv env
source env/bin/activate

종속성 설치

requirements.txt 파일을 만들고 다음을 추가합니다.

# requirements.txt
deepface

그런 다음 다음을 통해 설치하십시오.

pip install -r requirements.txt

소스 코드 작성

main.py 파일을 만들고 다음 모듈을 가져옵니다.

import argparse
from deepface import DeepFace

다음으로 메인 함수를 생성해야 합니다.

if __name__ == "__main__":
  ap = argparse.ArgumentParser()
  ap.add_argument("-i", "--image", required = True, help = "Path to input image")
  args = vars(ap.parse_args())

  img_path = args["image"]

  face_analysis = DeepFace.analyze(img_path = img_path)

  print ("gender:", face_analysis["gender"])
  print ("age:", face_analysis["age"])
  print ("dominant_race:", face_analysis["dominant_race"])
  print ("dominant_emotion", face_analysis["dominant_emotion"])

위의 코드는 이미지 파일을 받은 다음 이미지 파일을 분석을 위해 DeepFace 모듈에 전달합니다.

코드는 다음을 통해 실행할 수 있습니다.

python main.py -i lena.jpg

이 스크립트를 처음 실행하면 모델을 다운로드해야 하며 시간이 다소 걸립니다.

face_analysis를 모두 인쇄하면 다음과 같은 결과가 표시됩니다.

{'emotion': {'angry': 0.09911301312968135, 'disgust': 1.032224883346089e-06, 'fear': 2.6556044816970825, 'happy': 0.01839055767050013, 'sad': 65.46446681022644, 'surprise': 0.0007067909336910816, 'neutral': 31.761714816093445}, 'dominant_emotion': 'sad', 'region': {'x': 177, 'y': 77, 'w': 68, 'h': 68}, 'age': 31, 'gender': 'Woman', 'race': {'asian': 0.18712843253856495, 'indian': 0.08294145721779508, 'black': 0.007420518965146703, 'white': 90.12329519529911, 'middle eastern': 3.5380205385697208, 'latino hispanic': 6.061198178601156}, 'dominant_race': 'white'}

성별/나이/인종/감정의 출력을 인쇄하면 다음과 같은 출력이 표시됩니다.

gender: Woman
age: 31
dominant_race: white
dominant_emotion sad

다양한 자신의 이미지로 예제를 자유롭게 시도해보세요. 😎

결론

여기서 DeepFace 모듈을 소개했습니다. 내 자신의 모델 등을 교육한 경험이 있습니다. 하지만 이 모듈이 매우 유용하고 자신의 모델 등을 교육할 필요 없이 단 몇 줄의 코드로 사용할 수 있다고 생각했습니다.

자유롭게 사용해보고 다른 유용한 모듈 등이 있으면 알려주세요. 👀

코드는 https://github.com/ethand91/simple_deepface_example에서 찾을 수 있습니다.

행복한 코딩!


내가 일하는 것처럼? 다양한 주제에 대해 포스팅하고 있으니 더 보고 싶으시면 좋아요와 팔로우 부탁드립니다.
또한 저는 커피를 좋아합니다.