Python教程

python 图像等比例缩放

本文主要是介绍python 图像等比例缩放,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

图像等比例缩放

  • 图像缩放
    • 等比例缩放
      • pytorch下,对多维tensor进行缩放

图像缩放

目前比较常用的就是opencv-python底下的cv2.resize()

cv2.resize(src, size, fx, fy, interpolation)

其中src是输入原始图像,size为缩放的尺寸,fx为可选择的沿水平轴的比例因子,fy为可选择的沿垂直轴的比例因子,interpolation是可选择的插值方法,其中有:cv.INTER_NEAREST最近邻插值;cv.INTER_LINEAR双线性插值;cv.INTER_CUBIC三次样条插值;cv.INTER_AREA区域插值,默认是使用cv.INTER_LINEAR。

一般使用样例:

import cv2
# 原始图像读取
image = cv2.imread("XX.jpg")
# 输入你想要resize的图像尺寸。
size = 640
image_resize = cv2.resize(image, size)

目录下图像的遍历resize并保存:

import os
import cv2
# 输入原始图像存在的文件夹
datadir = "XXX"
# 设置保存路径
save_path = 'XXX'
if not os.path.exists(save_path):#如果路径不存在
	os.makedirs(save_path)
# 输入你想要resize的图像尺寸。
SIZE = 640
#使用os.path生成路径。
path = os.path.join(datadir)
#使用os.listdir返回path路径下所有图像文件。
img_list = os.listdir(path)
# 遍历图像文件
for i in img_list:
    img_array = cv2.imread(os.path.join(path, i), cv2.IMREAD_COLOR)
    # resize
    new_image = cv2.resize(img_array, (SIZE, SIZE))
    os.path.join(save_path, str(i)) # 保存的图片与原始图片同名
    #保存图片
    cv2.imwrite(save_path, new_image)

等比例缩放

在这里,我只总结一个我自己使用的cv2下的等比例resize。

同样,一般使用案例:

import cv2
# 原始图像读取
image = cv2.imread("XX.jpg")
# 输入你想要resize的图像高。
size = 640
# 获取原始图像宽高。
height, width = image.shape[0], image.shape[1]
# 等比例缩放尺度。
scale = height/size
# 获得相应等比例的图像宽度。
width_size = int(width/scale)
# resize
image_resize = cv2.resize(image, (width_size, size))

同样目录下使用:

import os
import cv2
# 输入原始图像存在的文件夹
datadir = "XXX"
# 设置保存路径
save_path = 'XXX'
if not os.path.exists(save_path):#如果路径不存在
	os.makedirs(save_path)
# 输入你想要resize的图像尺寸。
size = 640 
# 获取原始图像宽高。
height, width = image.shape[0], image.shape[1]
# 等比例缩放尺度。
scale = height/size    # 1
# 获得相应等比例的图像宽度。
width_size = int(width/scale)  # 2
#使用os.path生成路径。
path = os.path.join(datadir)
#使用os.listdir返回path路径下所有图像文件。
img_list = os.listdir(path)
# 遍历图像文件
for i in img_list:
    img_array = cv2.imread(os.path.join(path, i), cv2.IMREAD_COLOR)
    # resize
    new_image = cv2.resize(img_array, (width_size, size))
    os.path.join(save_path, str(i)) # 保存的图片与原始图片同名
    #保存图片
    cv2.imwrite(save_path, new_image)

我这里是按照修改高度,来等比例修改宽度的,如果你要先修改宽度,就调换一下1,2的位置就可以了。

pytorch下,对多维tensor进行缩放

我想对特征图进行缩放,它是tensor类型的。所以我们用不了cv2.resize,我看了网上有很多resize的方法,有说使用torch下的resize_(),用transform.resize()。我还是选择使用torch.nn.functional.interpolate
我亲测觉得对特征图缩放好用。

先来说一下这个函数。

torch.nn.functional.interpolate(input, size=None, scale_factor=None, mode='nearest', align_corners=None)

其中,input是我们的输入张量(Tensor);size是输出大小(int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int]);scale_factor (float or Tuple[float]) 指定输出为输入的多少倍数。如果输入为tuple,其也要制定为tuple类型;mode就是上采样算法了,有’nearest’, ‘linear’, ‘bilinear’, ‘bicubic’ , ‘trilinear’和’area’,其中默认是’nearest’;最后一个,我们用官方的解释:
align_corners (bool, optional) – 几何上,我们认为输入和输出的像素是正方形,而不是点。如果设置为True,则输入和输出张量由其角像素的中心点对齐,从而保留角像素处的值。如果设置为False,则输入和输出张量由它们的角像素的角点对齐,插值使用边界外值的边值填充;当scale_factor保持不变时,使该操作独立于输入大小。仅当使用的算法为’linear’, ‘bilinear’, 'bilinear’or 'trilinear’时可以使用。默认设置为False。

我列出了我一般的Tensor缩放。
这里要注意一点,这个函数要求我要四维张量。也就是B,C,W,H
所以一般使用案例:

# 我这里的tran_mask是一个三维的张量,所以我进行了一个.unsqueeze()的操作。补齐了四维
train_mask = train_mask.unsqueeze(1).float()
# 然后我采用的是四分之一的下采样操作。
train_mask = torch.nn.functional.interpolate(train_mask, scale_factor=1 / 4, mode='bilinear',
                                                    align_corners=False)
# 如果你们还要返回你之前的三维张量,那么就进行如下操作。.squeeze()。
train_mask = train_mask.squeeze(1).float()                                              

主要函数还是在于torch.nn.functional.interpolate(),这也就不列出什么目录下的下采样了。我们主要还是对特征层进行处理。当然,它resize的是每个通道下的W,H哦。

我觉得挺好用,之前用resize_的时候,根本不知道变成了啥,。。。。。。。。。。。

这篇关于python 图像等比例缩放的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!