以下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附近,比较满意。
2021-09-05 14:07:06 539.56MB 1、tensorflow 2、YOLOv4 3、objectdetecti
1