darknet-yolov4预标注

此代码在源代码darknet.py的基础上进行修改,将得到的输出结果写入xml文件中形成预标注文件,再使用lab
elImg在此基础上进行标注就会快速很多(主要是人工检查)。
以下是使用效果:
darknet-yolov4预标注_第1张图片
进度条显示总处理的图片数和预计剩余时间以及传输速度,最后输出平均每张图片预测的速度。(单位:s)
使用labelImg打开,查看效果:
官方的yolov4.weights模型检测效果看是挺准确的。

------------------------分割线7.29-------------------------------

后更新的代码解决遇到没有目标而引起的批量检测异常中断问题,以及可以自定义标签,只预标注感兴趣的类别标签,屏蔽模型其他的类别标注

你可能感兴趣的:(深度学习,python)