文件名称:YOLOv4_VOC.zip
文件大小:539.56MB
文件格式:ZIP
更新时间:2024-09-24 13:19:01
1、tensorflow 2、YOLOv4 3、objectdetecti
以下4个py文件需要单独运行: 首先运行annotation.py文件:将VOC数据集进行划分,生成.txt路径文件存储到model_data文件夹中。 然后运行k_means_calculate.py文件:计算生成anchors数值,存储到model_data文件夹中。 再然后运行train.py文件:加载原始权重,训练YOLOv4模型,并将每轮训练的结果存储进Logs文件夹中。 最后运行yolo_predict.py文件:载入训练好的YOLOv4权重,对测试集数据进行检测,检测结果存放入demo文件夹中。 YOLOv4每张图片检测耗时373.5ms,精度较高,训练30 epoch左右后val loss降低至2.3附近,比较满意。