摘要

火电厂生产过程中存在油污泄漏污染雨排口的情况,不及时处理会污染河流环境。雨排口监控图像为俯角图像,用矩形检测框难以正确识别,基于某电厂雨排口俯角图像,提出了一种平行四边形检测框算法。分析了平行四边形检测框的特性和角度检测误差对交并比(IoU)的影响,讨论了分类和回归两种角度检测方法的特点;对于角度分类问题,设计了基于指数函数的类平滑标签;针对类别间隔角度的检测,设计了分类-回归器,学习角度类别间隔的偏移量。最后使用改进后的分类交叉熵损失函数和IoU-Smooth L1损失进行训练。在电厂雨排口数据集中测试表明,所提方法检测效果最好。

全文