OpenCV的resize方法與雙線性插值

訓練Object Detection模型SSD完畢之后進入test階段,每張圖像在進入輸入層之前需要進行resize操作,以滿足CNN模型對輸入層size的要求。本文首先介紹了Caffe實現的SSD模型對輸入圖像的變換規定,引出了OpenCV中的resize方法,最后介紹該方法中的插值參數cv.INTER_LINEAR和該插值方法的原理。

caffe_ssd

caffe_ssd在test階段,對圖像的變換設置如下:

test_transform_param = {
        'mean_value': [104, 117, 123],
        'force_color': True,
        'resize_param': {
                'prob': 1,
                'resize_mode': P.Resize.WARP,
                'height': resize_height,
                'width': resize_width,
                'interp_mode': [P.Resize.LINEAR],
                },
        }

以上設定來自ssd_coco.py

  • 'mean_value': [104, 117, 123]是ImageNet圖像BGR三個通道的均值。每張圖像分別需要減去相應通道的均值,實現中心化。
  • 'force_color': True強制采用彩色BGR圖像模式,防止灰度圖像維度與SSD模型輸入層維度不一致。
  • resize_param屬性在caffe.proto的ResizeParameter中有說明
    // Message that stores parameters used by data transformer for resize policy
    message ResizeParameter {
      //Probability of using this resize policy
      optional float prob = 1 [default = 1];
    
      enum Resize_mode {
        WARP = 1;
        FIT_SMALL_SIZE = 2;
        FIT_LARGE_SIZE_AND_PAD = 3;
      }
      optional Resize_mode resize_mode = 2 [default = WARP];
      optional uint32 height = 3 [default = 0];
      optional uint32 width = 4 [default = 0];
      // A parameter used to update bbox in FIT_SMALL_SIZE mode.
      optional uint32 height_scale = 8 [default = 0];
      optional uint32 width_scale = 9 [default = 0];
    
      enum Pad_mode {
        CONSTANT = 1;
        MIRRORED = 2;
        REPEAT_NEAREST = 3;
      }
      // Padding mode for BE_SMALL_SIZE_AND_PAD mode and object centering
      optional Pad_mode pad_mode = 5 [default = CONSTANT];
      // if specified can be repeated once (would fill all the channels)
      // or can be repeated the same number of times as channels
      // (would use it them to the corresponding channel)
      repeated float pad_value = 6;
    
      enum Interp_mode { //Same as in OpenCV
        LINEAR = 1;
        AREA = 2;
        NEAREST = 3;
        CUBIC = 4;
        LANCZOS4 = 5;
      }
      //interpolation for for resizing
      repeated Interp_mode interp_mode = 7;
    }
    
    其中的interp_mode采用LINEAR模式對圖像進行Resize操作,與Opencv中的resize一致。

接下來,我們具體介紹一下OpenCV中的resize方法。

resize方法的簽名

C++:   
void cv::resize (InputArray src,
    OutputArray             dst,
    Size                      dsize,
    double                  fx = 0,
    double                  fy = 0,
    int                       interpolation = INTER_LINEAR
)       
Python:   
dst =   cv.resize(src, dsize[, dst[, fx[, fy[, interpolation]]]])

參數說明:

src 輸入圖像.
dst 輸出圖像; 其size為dsize,或由src.size()、fx與fy計算而得; dst類型與src保持一致.   
dsize 輸出圖像的size; 如果設為0,或(0, 0), 計算方式為:
    ?????????? = ????????(??????????(????*??????.????????), ??????????(????*??????.????????))
dsize和(fx, fy)必須有一組不為0. 如果都為0,無法確定被resize后的圖像大小
fx  水平軸縮放因子; 等于0時,計算方式為:
    (????????????)??????????.??????????/??????.????????
fy  豎直軸縮放因子; 等于0時,計算方式為:
    (????????????)??????????.????????????/??????.????????
interpolation   差值方法, 方法見InterpolationFlags

InterpolationFlags
縮小圖像時,一般INTER_AREA插值效果較好。放大圖像時, INTER_CUBIC (slow)更好些,或INTER_LINEAR (faster but still looks OK)。

Resize Image Example

import cv2 as cv
img = cv.imread('./lena.jpg')
h, w = img.shape[:2]
# 縮小圖像到原來一半大小,方法一,設置dsize
dst = cv.resize(img, (h//2, w//2), None, 0, 0, cv.INTER_LINEAR)
cv.imwrite('./lena1.jpg', dst)
# 縮小圖像到原來一半大小,方法二,設置fx和fy
dst = cv.resize(src=img, dsize=None, fx=0.5, fy=0.5, interpolation=cv.INTER_LINEAR)
cv.imwrite('./lena2.jpg', dst)

cv.INTER_LINEAR的原理

resize方法提供了9種插值參數,Caffe中支持的5種分別是

  • 最近鄰插值法 cv.INTER_NEAREST
  • 雙線性插值法 cv.INTER_LINEAR
  • 雙三次插值法 cv.INTER_CUBIC
  • 區域插值法 cv.INTER_AREA
  • 蘭索斯插值法 cv.INTER_LANCZOS4

下面具體介紹一下雙線性插值法,這種插值方法最易于理解,也應用最多。

首先,在x軸方向進行線性插值:藍色點得到綠色點
x軸方向線性插值

然后,在y軸方向進行線性插值:綠色點得到橙色點
y軸方向線性插值

最后,簡化為矩陣變換的形式:
雙線性插值的矩陣變換表達
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 229,001評論 6 537
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,786評論 3 423
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,986評論 0 381
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,204評論 1 315
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,964評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,354評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,410評論 3 444
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,554評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,106評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,918評論 3 356
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,093評論 1 371
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,648評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,342評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,755評論 0 28
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 36,009評論 1 289
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,839評論 3 395
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,107評論 2 375

推薦閱讀更多精彩內容