TensorFlow Models 사용 방법

공개되고 있는 TensorFlow Models 의 사용법입니다.

프로그램 다운로드

git clone https://github.com/tensorflow/models.git

샘플 실행

cd models/tutorials/image/imagenet
python classify_image.py

이 프로그램은/tmp/imagenet/cropped_panda.jpg를 처리합니다. 팬더로 판단됩니다.

큰 데이터가/tmp/imagenet 이하로 다운로드됩니다.
/tmp는 재부팅으로 지워지므로/var/tmp/imagenet으로 이동합니다.

이에 따라 classify_image.py도 수정합니다.
default='/tmp/imagenet',
을
default='/var/tmp/imagenet',
합니다.

예를 들어 딸기 이미지를 다운로드하고 테스트하면

$ export TF_CPP_MIN_LOG_LEVEL=2
$ python classify_image.py --image_file  ~/Downloads/strawberry.jpg 
strawberry (score = 0.97967)
banana (score = 0.00068)
Leonberg (score = 0.00047)
ringneck snake, ring-necked snake, ring snake (score = 0.00047)
German shepherd, German shepherd dog, German police dog, alsatian (score = 0.00046)

제대로 strawberry (score = 0.97967) 라고 판정합니다.

바나나 이미지로 시도하면

$ export TF_CPP_MIN_LOG_LEVEL=2
$ python classify_image.py --image_file  ~/Downloads/banana4.jpg
banana (score = 0.99953)
orange (score = 0.00002)
pineapple, ananas (score = 0.00001)
lakeside, lakeshore (score = 0.00001)
corn (score = 0.00000)

제대로, banana (score = 0.99953)라고 판정합니다.

그렇지만 사과의 성적은 그다지 좋지 않습니다.

$ export TF_CPP_MIN_LOG_LEVEL=2
$ python classify_image.py --image_file  ~/Downloads/apple2.jpg 
orange (score = 0.23259)
Granny Smith (score = 0.22257)
banana (score = 0.05309)
pomegranate (score = 0.04968)
strawberry (score = 0.01616)

서양산 사과가 아니면 좋은 결과가 나오지 않을지도 모릅니다.

다음으로 동작 확인했습니다.

Arch Linux (4.15.15-1-ARCH)
Raspbian GNU/Linux 9 (stretch)
Raspbian GNU/Linux 8 (jessie)