科研之友
微信
新浪微博
Facebook
分享链接
摘要
目前目标检测算法由于数量和背景不平衡导致的电力部件检测精度不高,同时在背景复杂的情况下分类定位不准确。针对上述问题,提出一种改进的YOLOv3的变压器检测算法,使用DarkNet-53深度卷积神经网络作为抽取图像特征的骨干网络。使用Focal loss和均衡交叉熵函数改进原YOLOv3的损失函数,使模型在样本数量少的类别上倾注更多的注意力;使用K-means对样本中的变压器框进行聚类生成预选框,并调整YOLOv3输出结构使其适应变压器目标检测任务,提高检测的效率。实验结果表明改进后的YOLOv3在变压器检测上有更高的准确率和速率。
- 单位
