节点文献
基于残差网络和改进特征金字塔的油田作业现场目标检测算法
Field Object Detection for Oilfield Operation Based on Residual Network and Improved Feature Pyramid Networks
【摘要】 针对单点多盒检测器(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可以达到实时处理的要求。
【Abstract】 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.
【Key words】 deep learning; single shot multibox detector(SSD); small object detection; feature pyramid networks; residual network; dilated convolution; oilfield security;
- 【文献出处】 科学技术与工程 ,Science Technology and Engineering , 编辑部邮箱 ,2020年11期
- 【分类号】TP391.41;TP18;TE319
- 【被引频次】6
- 【下载频次】205