MS COCO Pretrained Model을 사용하여 고양이와 개 주변에 Bounding Box를 그려 봅니다.
- 필요한 package들을 import합니다.
- models/research를 download하고 path를 설정한 후 필요한 package들을 import합니다.
- model이름을 MODEL_NAME에 저장합니다.
- model을 사용하기 위해 필요한 값들을 설정합니다.
- model을 download하고 압축을 풉니다.
- url에 있는 image를 load하는 함수를 정의합니다.
- color값은 0.0~1.0을 가지도록 합니다.
- alpha channel은 사용하지 않습니다.
- image를 그리는 함수를 정의합니다.
- image에서 label과 bounding box정보를 얻어오는 함수를 정의합니다.
- object detection network의 계산과정(graph)을 load합니다.
- label을 사람이 읽을 수 있는 형태로 바꿔주는 dictionary를 생성하여 category_index에 저장합니다.
- 각종 image url을 가지고 object detection network에 넣어서 인식결과를 살펴봅니다.
토론이 없습니다