科研之友
微信
新浪微博
Facebook
分享链接
摘要
为实现接触轨表面缺陷的高效检测，提出了一个基于改进YOLOv3模型的缺陷检测方法。一方面，采用轻量化的Ghost组件代替原始YOLOv3模型Darkent-53网络中的残差模块，来减少模型的尺寸和提高推理速度。另一方面，使用GIoU解决IoU在检测框和真值框不相交时损失函数无法反向传播的问题，并使用GIoU损失作为边界框损失加快模型收敛速度。利用在上海地铁16号线采集到的接触轨图像进行实验，精确率和召回率达到90.21%和89.64%，检测速度达到每张图片0.015 s，模型尺寸压缩到59.5 MB。
-
单位上海地铁维护保障有限公司供电分公司
