Python 활용 딥러닝
CNN으로 꽃 이미지 예측하기
Penguin's Repository
2018. 12. 20. 14:14
아래와 같은 폴더 구조로 만든다(GoogleNet/workspace)
predict.py
readme.txtreadme.txt 파일을 참고하여
이미지 파일을 다운로드 받고, Training을 위해 학습 명령어를 Terminal에 입력한다.
1 | python retrain.py --bottleneck_dir=./workspace/bottlenecks --model_dir=./workspace/inception --output_graph=./workspace/flowers_graph.pb --output_labels=./workspace/flowers_labels.txt --image_dir ./workspace/flower_photos --how_many_training_steps 1000 | cs |
Training이 완료되면 Terminal에서 아래와 같이 predict.py 파일을 실행해서 예측을 해본다.
1 | python predict.py ./workspace/flower_photos/daisy/267148092_4bb874af58.jpg | cs |