keras-yolo3-master

Posted 2021-01-17 gloria11

logs/000/trained_weights_final.h5 放置训练完的权重

  

keras-yolo3-master
Keras/Tensorflow+python+yolo3训练自己的数据集
代码：https://github.com/qqwweee/keras-yolo3

修改yolov3.cfg文件：https://blog.csdn.net/lilai619/article/details/79695109

使用yolo3训练自己的数据集进行目标检测
VOCdevkit/VOC2007/Annotations xml文件
VOCdevkit/VOC2007/JPEGImages   jpg图片
VOCdevkit/VOC2007/ImageSets/Main 下面四个文件，在VOC2007下创建文件test.py,
运行voc_annotation.py,修改classes,运行结果是2007_train.txt,手动删除2007_,生成train.txt,


修改参数文件：yolo3.cfg

三处yolo 各修改filters:3*(5+len(classes));
               classes:len(classes)
               random:原来是1,显存小改为0,


修改model_data下文件，coco_classes.txt：放入类名 voc_classes.txt：放入类名，注意不要有空行或空格

训练：
python convert.py -w yolov3.cfg yolov3.weights model_data/yolo_weights.h5
python train.py

测试：
python yolo.py   或者python yolo_video.py video_path