Journal of Jilin University Science Edition ›› 2024, Vol. 62 ›› Issue (3): 643-654.

Previous Articles     Next Articles

A Region of Interest Pooling Algorithm for Edge Gradient Interpolation

ZHOU Yuejin1,2, DING Jiayi1   

  1. 1. School of Mathematics and Big Data, Anhui University of Science and Technology, Huainan 232001, Anhui Province, China; 
    2. State Key Laboratory of Mining Response and Disaster Prevention and Control in Deep Coal Mines, Huainan 232001, Anhui Province, China
  • Received:2023-06-02 Online:2024-05-26 Published:2024-05-26

Abstract: Aiming at the problems that the existing mainstream target detection algorithms had  low detection accuracy and incomplete segmentation in the  image edge regions, we proposed a region of interest pooling algorithm based on Mask RCNN model. Firstly, the feature maps of the regions of interest were divided into edge regions and non-edge regions by the Otsu threshold segmentation method. Secondly, the edge gradient interpolation algorithm was used to interpolate for the edge regions, 
and the bilinear interpolation algorithm was used to interpolate for the non-edge regions so that the discrete feature map was mapped into a continuous space. Thirdly,  the interpolated feature maps were evenly divided into k×k units. Finally, the double integral was used to calculate the average value of each unit to complete the pooling operation. The comparative experimental results show that the proposed algorithm, based on the Mask RCNN model, has a certain improvement in detection accuracy  compared with existing algorithms on COCO(2014) dataset, and has a good segmentation effect on the details of the image edge regions.

Key words: Mask RCNN model, region of interest pooling, Otsu threshold segmentation, edge gradient interpolation, bilinear interpolation

CLC Number: 

  • TP391