一、所需要的库文件以及Python版本
- Python 3.7 + 开发环境
- TensorFlow-gpu 2.6.0
- OpenCV-Python 4.5.3+
- win10
- Microsoft Visual Studio 2019(我反正要下,缺少这个,因为这个弄得我很难受)
二、下载YOLO3源代码
- YOLO3源代码:github_keras-yolo3
- 下载权重文件:官网_yolov3_weights(点击直接下载)
三、添加YOLO模型
先将刚刚下载好的yolov3.weights 权重放在下载的yolo文件中
再用Pycharm打开YOLO源代码
在命令行输入:python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
运行命令直到出现:Saved Keras model to model_data/yolo.h5
且 model_data 中多了个 yolo.h5 文件就代表准备工作完