一、转换过程
1、下载文件yolo.weights:
2、下载文件yolov2.cfg,可改名成yolo.cfg
https://github.com/pjreddie/darknet/blob/master/cfg/yolov2.cfg
3、下载需要脚本
https://github.com/allanzelener/YAD2Khttps://github.com/allanzelener/YAD2K
4、复制yolo.weights
和yolo.cfg
以及yad2k.py
三个文件,以及一个文件夹yad2k
到单独的文件夹
ps: yad2k.py和yad2k在YAD2K文件夹中:
效果如下:
如何直接在Anocnoda Prompt中输入:
python yad2k.py yolov.cfg yolov3.weights model_data/yolo.h5
运行结束,打开model_data即可发现:
二、出现问题
1、版本问题
注意tensorflow 和keras版本问题,不能太高 ,否则会一直报错
可以创建虚拟环境后,在安装以下版本:
pip uninstall tensorflow keras
pip install tensorflow==1.14.0 keras==2.2.4
ps : 记住暂时关闭用户代理,否则,安装不上
2、可能提示没有model_data文件夹
需要手动先创建一个