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

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

ria family span soft ast scale pat style 操作

在使用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];

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