Yet Another Darknet 2 Keras
- 参考:https://github.com/allanzelener/YAD2K
- 该项目使用了keras与tensorflow-gpu,因此可以在任何使用该框架的环境下运行
使用
- 下载模型
- 将 Darknet YOLO_v2 model转换为Keras model
- python yad2k.py yolo.cfg yolo.weights model_data/yolo.h5
- only supports YOLO_v2 style models, this include the following configurations: darknet19_448, tiny-yolo-voc, yolo-voc, and yolo
- 测试
- python test_yolo.py model_data/yolo.h5
- 图片位于 images/文件夹
- 输出结果在images/out/文件夹
- python test_yolo.py model_data/yolo.h5
- 问题
- 将输出需结果目录改为out
- 若想显示中文需将字体改为中文字体,C:\Windows\Fonts\simkai.ttf
- 更改YAD2K-master\model_data\coco_classes.txt对应类别名称