參考:python圖像大小縮放使用cv2.resize()或scipy.ndimage.zoom()
參考:【Scipy】scipy.ndimage.zoom矩陣放縮
數據在輸入到 U-Net 網絡里面,尺寸需要是 32 的倍數,這樣才能保證輸入與輸出尺寸一致。對於普通圖片而言,可以直接通過 PIL 或者 OpenCV 提供的 resize 函數來進行相應的操作。但是對於 geotiff 數據而言,其數值為 uint16,如果轉為圖像的話會有相應的耗損,下面介紹直接對 numpy.array 進行縮放的操作。
-
zoom(input, zoom, ...)
-
input: 輸入多維矩陣;
-
zoom: float/sequence,沿軸的縮放系數,如果是浮點型,表示每個軸的縮放是相同的,如果是序列,zoom應包含每個軸的縮放值;
舉例:
# cat_cubic_arr 尺寸 (250, 250, 3) # 將其轉為 (190, 190, 3) # 對不同的軸進行縮放如下 cat_cubic_arr3 = zoom(cat_cubic_arr, (190/250, 190/250, 1))