[个人笔记]使用keras-yolo3训练自己的数据集

1、下载源码

从这里下载yolov3的keras实现源码

git clone https://github.com/qqwweee/keras-yolo3.git

2、使用预训练模型测试

下载yolov3原作者C语言实现预训练模型yolov3.weights，然后放到keras-yolo3目录下。

然后在keras-yolo3目录下运行

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

将权重参数转成h5格式。

运行

python yolo_video.py --image

然后会提示

Input image filename:

输入图像文件路径，就会显示检测结果