[Project] 닮은 꼴 연예인 찾기(face_recognition)

face_recognition

• 얼굴 인식 라이브러리
• cmake, opencv, dlib를 사전에 설치한 후 라이브러리 설치 가능

구현 방법

• Anaconda Prompt 실행 후 가상환경 생성

conda create -n test(가상환경 이름)
conda activate test(가상환경 이름)

• face_recognition 및 필요 라이브러리 설치

conda install numpy
conda install pillow
conda install opencv
conda install scipy
conda install cmake
conda install -c conda-forge dlib
conda install -c akode face_recogntion_models

코드 구현

• 연예인 일부 사진을 다운 받은 후 images폴더에 저장
• "0.jpg" 라는 비교하고 싶은 대상 사진 저장

import face_recognition
import os
import matplotlib.image as img
import matplotlib.pyplot as pp입력하세요

images = os.listdir('images')
print(images)

image_to_be_matched = face_recognition.load_image_file('0.jpg')
image_to_be_matched_encoded = face_recognition.face_encodings(image_to_be_matched)[0]

• 이미지 비교 모델

for image in images:
    current_image = face_recognition.load_image_file("images/" + image)
    current_image_encoded = face_recognition.face_encodings(current_image)[0]
    result = face_recognition.compare_faces([image_to_be_matched_encoded], current_image_encoded)
    if result[0] == True:
        print("Matched: " + image)
        fileName = "images/" + image
        ndarray = img.imread(fileName)
        pp.imshow(ndarray)
        pp.show()
        break
    #else:
        #print("Not matched: " + image)