ImageNet Pretrained Model을 사용하여 Image의 유사도를 구해 봅니다.
- 필요한 package들을 import합니다.
- models/research/slim download하고 path를 설정한 후 필요한 package들을 import합니다.
- url에 있는 image를 load하여 image_size * image_size의 크기로 resize하는 함수를 정의합니다.
- color값은 0.0~1.0을 가지도록 합니다.
- alpha channel은 사용하지 않습니다.
- image를 그리는 함수를 정의합니다.
- model이름을 model_name에 저장합니다.
- model별로 적절하게 checkpoint, num_classes, labels_offset, pixel_range를 설정해 주고 imagenet pretrained model checkpoint를 download한 후 압축을 풉니다.
- cnn의 계산과정(graph)을 정의합니다.
- Tensorflow Session을 open합니다.
- checkpoint로부터 parameter들을 load하여 tensorflow variable들을 초기화합니다.
- 각종 image url 쌍을 가지고 cnn에 넣어서 image의 유사도를 살펴봅니다.
- Tensorflow Session을 close합니다.
토론이 없습니다