[python] 뽀로로 친구들을 검출해보자(Tensorflow object detection API)

코딩/python|2020. 6. 19. 17:28

뽀로로, 크롱, 루피를 검출할 수 있는 객체 검출기(object detector)를 직접 훈련시켜 만들어봤습니다. 보시다시피 뽀로로와 크롱과 루피를 나름대로 잘 검출하죠?  

 

 

훈련하는데 많은 수의 이미지를 활용하진 않았기 때문에 완벽하진 않습니다.

 

 

훈련하는데 사용하지 않은 뽀로로펜도 뽀로로로 검출이 되더군요. 

 

 

이렇게 사진 속 친구들도 꽤 잘 검출됩니다. 

 

 

저는 이 검출기를 Tensorflow object detection API를 이용해서 만들었습니다. 저는 이 모델을 만드는데 있어서 처음부터 끝까지 Gilbert Tanner라는 분의 영상과 블로그 글들을 참고했습니다. 정말 쉽게 잘 설명해놓으셨습니다. 그냥 따라하기만 하면 됩니다. 물론 시간은 꽤 걸리지만요. 

 

www.youtube.com/playlist?list=PLoY9f-5DCBlkvkvDjwa5Dn4WXjhTA5K7v

 

Tensorflow Object Detection - YouTube

Collection of all my Tensorflow Object Detection API tutorials.

www.youtube.com

 

자신만의 객체 검출기를 한번 만들어보고 싶으신 분은 이 분의 영상들과 글을 참고해서 해보시길 추천드립니다.