1. 程式人生 > >在CNN網路中roi從原圖對映到feature map中的計算方法

在CNN網路中roi從原圖對映到feature map中的計算方法

在使用fast rcnn以及faster rcnn做檢測任務的時候,涉及到從影象的roi區域到feature map中roi的對映,然後再進行roi_pooling之類的操作。
比如影象的大小是(600,800),在經過一系列的卷積以及pooling操作之後在某一個層中得到的feature map大小是(38,50),那麼在原圖中roi是(30,40,200,400),
在feature map中對應的roi區域應該是
roi_start_w = round(30 * spatial_scale);
roi_start_h = round(40 * spatial_scale);
roi_end_w = round(200 * spatial_scale);
roi_end_h = round(400 * spatial_scale);
其中spatial_scale的計算方式是spatial_scale=round(38/600)=round(50/800)=0.0625,所以在feature map中的roi區域[roi_start_w,roi_start_h,roi_end_w,roi_end_h]=[2,3,13,25]