Coding/Python
[Python] zAI 라이브러리를 이용하여 간단한 인공지능 사용하기 - 3
junedev
2019. 10. 10. 10:46
zAI 라이브러리 - 사람 얼굴 추출하기
이전 포스팅에서 라이브러리 설치와 API Key 발급을 마쳤습니다.
zAI 라이브러리를 이용하여 간단한 인공지능 사용하기 - 1(라이브러리 설치)
zAI 라이브러리를 이용하여 간단한 인공지능 사용하기 - 2(API Key 발급)
이제 zAI 라이브러리를 실제로 사용해봅시다.
우선, 사람이 여러 명 있는 사진을 준비합시다.
import zAI
from zAI import zImage
zAI.utils.set_backend_key(key_name='MICROSOFT_AZURE_VISION_API_KEY', key_value='API KEY', save=True)
zAI.utils.set_backend_key(key_name='MICROSOFT_AZURE_URL', key_value='END POINT', save=True)
imageLink = 'people.jpg'
image = zImage(imageLink)
image.find_faces(backend='local')
myCloseup = image.extract_face(n=3, margin=15) # margin is the number of pixels we will expand
myCloseup.display()
myCloseup.save("./face.jpg")
zAI 라이브러리로부터 zImage를 함께 import 해줍시다.
zImage는 zAI에서도 이미지를 다루는 기능이 있는 라이브러리 입니다.
key_name에 API Key와 Endpoint를 각각 넣어줍시다.
zImage 객체에 위의 이미지를 넣어주고, find_faces 메소드를 통해 로컬 환경에서 얼굴을 찾습니다.
extract_face는 그렇게 찾은 얼굴을 추출합니다.
위의 사진의 경우 얼굴이 총 4개 추출될 것입니다. n값은 그중 몇번째 얼굴을 저장할 것인지를 정하는 것입니다.
지금같은 경우 myCloseup에는 세번째 사람의 얼굴이 저장되겠네요.
display는 그렇게 얻어낸 이미지를 GUI환경에서 보여주는 메소드입니다.
이 코드를 돌려보면,
이렇게 사람의 얼굴이 추출된 이미지가 저장됩니다.
zAI 라이브러리를 이용하면 간단하게 사람의 얼굴만을 추출할 수 있군요