参考: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))