1.下载权重文件
Keras YOLO 需要 .h5格式的文件,转换方法请见
2.修改YOLO配置文件
修改分类,例如:
修改 'model_data/voc_classes.txt '中:
在 yolo.py 中修改配置内容。
_defaults = {
"权重文件.h5’’,
“anchors_path”: ‘model_data/yolo_anchors.txt’,
“classes_path”: ‘model_data/voc_classes.txt :使用修改过的分类’,
“score” : 0.3,
“iou” : 0.45,
“model_image_size” : (416, 416),
“gpu_num” : 1,
}
启动运行:
python yolo_video.py