TensorFlow + Keras YOLO V3初试及问题解决

 

YOLO V3的图像识别，一直想试一下。

实际参考了https://www.cnblogs.com/tensorflownews/p/8922359.html，并有各种百度及google

整理起来，简单来说分成以下几步：

0：环境的安装，包括pyhon3.x，keras这些

1：下载keras-yolo3的整个工程 https://github.com/qqwweee/keras-yolo3

2：从 YOLO 官网下载 YOLOv3 权重

wget https://pjreddie.com/media/files/yolov3.weights

3：转换 Darknet YOLO 模型为 Keras 模型

python3 convert.py yolov3.cfg yolov3.weights model_data/yolo.h5

4.运行YOLO 目标检测，按照提示输入要测试的图片即可

python3 yolo.py

 

遇到的问题：

1,最后一步执行的时候，提示出错：

File "yolo.py", line 21, in 

from keras.utils import multi_gpu_model
ImportError: cannot import name 'multi_gpu_model'

 原因是版本的匹配问题，将keras升级到最新的就OK了

pip3 install git+git://github.com/fchollet/keras.git --upgrade

2.提示cv2找不到

ModuleNotFoundError: No module named 'cv2'

以下命令安装opencv可以解决

pip3 install opencv-python