前提環境
前回の環境構築で準備。
- Windows10 Professional 64bit
- Anaconda3 5.0.0 64bit(Python3.6)
- CUDA 9.0
- cuDNN 7.1.3
- TensorFlow
- Keras
導入するもの
- YOLOv3 Keras実装
YOLOv3を取得
qqwweee/keras-yolo3でYOLOのKeras実装が公開されているので、cloneしてきました。
Quick Start
基本、READMEの通りに動かせばすぐに実行できました。
- YOLO websiteから学習済みパラメータを取得。
- yolo.h5を作る
- pillowが入ってなかったので入れた
- 適当な画像を用意し、yolo.pyを実行
実行結果
Found 22 boxes for img
truck 0.38 (98, 59) (109, 72)
truck 0.89 (54, 110) (81, 138)
bus 0.93 (37, 86) (67, 116)
car 0.30 (86, 92) (98, 99)
car 0.31 (88, 98) (98, 107)
car 0.43 (181, 84) (197, 94)
car 0.55 (103, 96) (119, 105)
car 0.55 (121, 66) (130, 73)
car 0.64 (84, 94) (99, 106)
car 0.66 (108, 67) (117, 74)
car 0.66 (169, 88) (181, 98)
car 0.80 (198, 96) (213, 106)
car 0.87 (169, 133) (198, 160)
car 0.91 (13, 145) (57, 186)
car 0.92 (149, 93) (165, 105)
car 0.95 (0, 127) (18, 159)
car 0.97 (181, 97) (199, 111)
car 0.97 (101, 107) (122, 120)
car 0.98 (111, 85) (127, 97)
car 0.98 (73, 101) (92, 116)
car 0.98 (88, 129) (115, 154)
car 0.99 (215, 163) (259, 193)
5.983695983886719