基于残差网络和改进特征金字塔的油田作业现场目标检测算法  

Field Object Detection for Oilfield Operation Based on Residual Network and Improved Feature Pyramid Networks

作　　者：梁鸿[1] 李洋[1] 邵明文[1] 李传秀[1] 张兆雷 LIANG Hong;LI Yang;SHAO Ming-wen;LI Chuan-xiu;ZHANG Zhao-lei(College of Computer Science and Technology,China University of Petroleum(East China),Qingdao 266000,China)

机构地区：[1]中国石油大学(华东)计算机科学与技术学院,青岛266000

出　　处：《科学技术与工程》2020年第11期4442-4450,共9页Science Technology and Engineering

基　　金：国家自然科学基金(61673396);2018年中央高校基本科研业务费项目(18CX02138A)。

摘　　要：针对单点多盒检测器(single shot multibox detector,SSD)对小目标识别率低的问题,提出一种基于残差网络和改进特征金字塔(feature pyramid networks,FPN)的RP-SSD(residual and pyramid SSD)算法,并将其应用于油田安防领域。为了得到小物体更多的信息,首先在特征金字塔中增加上采样模块,并在上采样模块之后添加预测模块,之后采用空洞卷积增大Conv43的感受野。RP-SSD网络变深,针对RP-SSD在反向传播过程中存在梯度爆炸或梯度消失的问题,采用跳层连接的方式改进基础网络。RP-SSD在PASCAL VOC测试的准确率(meanaverage precision,mAP)为78.9%,比SSD提高了1.7%,其中对于目标较小的bottle类提高了8.9%。实验结果表明,RP-SSD对小目标检测的性能提高显著,同时RP-SSD在GTX 1080Ti上测试的速度为32帧/s,可见RP-SSD可以达到实时处理的要求。To solve the problem of low recognition rate of small objects by single shot multibox detector(SSD), a RP-SSD(residual and pyramid SSD) algorithm was proposed based on residual network and improved feature pyramid networks. The RP-SSD was applied to the oil field security. To acquire more information of small objects, the upper sampling module was added to the feature pyramid networks first, and the prediction module was added after the upper sampling module. Then, the receptive field of Conv43 was increased by using dilated convolution, and the RP-SSD network was deepened. To solve the problem of gradient explosion or gradient disappearance in the back propagation process of RP-SSD, the basic network was improved by means of jump-layer connection. The mAP(mean average precision) of RP-SSD in PASCAL VOC test was 78.9%. RP-SSD was 1.7% higher than mAP of SSD, and 8.9% higher for bottle class with smaller object. The experimental results show that the performance of RP-SSD for small object detection was significantly improved. At the same time, the speed of RP-SSD testing on GTX 1080 Ti was 32 fps(frame per second). Therefore, the RP-SSD could meet the requirements of real-time processing.

关 键 词：深度学习 单点多盒检测器(SSD) 小目标检测 特征金字塔 残差网络 空洞卷积 油田安防 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]