resize训练集图片大小并存储的方法

我们做深度学习训练时,需要大量的图片(数据),这些图片往往尺寸不一,也不是我们input的尺寸需求。如果我们能在训练前就将这些图片resize成需要的尺寸,那么训练过程将会非常省时省资源的。

resize图片大小的方法有两种:

一、#直接基于原图片中心点为中心,以原图片较短边切割成边长为原长方形较短边为边长的正方形

代码如下:

'''
[1]  #直接基于原图片中心点为中心,以原图片较短边切割成边长为原长方形较短边为边长的正方形
#如test.jpg和crop_img.jpg,部分长宽比例较大的图形经过切割可能会丢失重要的信息,故一般不用!

import matplotlib.image as mpimg

def load_image(path):
    # 读取图片,rgb
    img = mpimg.imread(path)
    # 将图片修剪成中心的正方形
    short_edge = min(img.shape[:2])
    yy = int((img.shape[0] - short_edge) / 2)
    xx = int((img.shape[1] - short_edge) / 2)
    crop_img = img[yy: yy + short_edge, xx: xx + short_edge]
    mpimg.imsave('crop_img',crop_img)
    return crop_img

load_image('./test.jpg')
'''

输入图片
resize训练集图片大小并存储的方法
输出图片
resize训练集图片大小并存储的方法
显而易见:如果按照方法一做简单的切割,很容易丢失大量有用的信息,导致训练正确率低,甚至出现错误,故我们一般不用这种方法

二、#推荐的resize并存储的方法如下(双线性插值,或者双三次插值)

代码如下

[2]#推荐的resize并存储的方法如下

'''
1、我们将搜集到的图片存放在与软件脚本同一根目录下的'./data/image/pre_train/'文件夹下;
2、我们使用Image模块从pre_train文件夹读取图片,resize,并存储入'./data/image/train/'文件夹;
'''

from PIL import Image
import os

src_path = './data/image/pre_train/'
dst_path = './data/image/train/'

filelist=os.listdir(src_path)

for img in filelist:
    image=Image.open(src_path+img)
    image_resize=image.resize((224,224),resample=2)
    #image.resize(size,resample=0)  #sesam用于表示改变图像过程中的插值方法,0:双线性插值;1:最邻近插值;2:双三次插值;3|面积插值法
    #参考:python: PIL的Image.resize()函数:
    image_resize.save(dst_path+img)

还是那只猫,我们看看这次处理后的效果
resize训练集图片大小并存储的方法
还是一眼就认出这是只猫,对不对?!

我们在对比下处理前后的数据集
resize训练集图片大小并存储的方法
resize训练集图片大小并存储的方法

参考资料:
1、Python:PIL的 Image.resize()函数:https://blog.csdn.net/weixin_40522801/article/details/106491212?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522161703255516780266279470%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=161703255516780266279470&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v29-1-106491212.first_rank_v2_pc_rank_v29&utm_term=python%3A+PIL%E7%9A%84Image.resize%28%29%E5%87%BD%E6%95%B0&spm=1018.2226.3001.4187
2、python 转换图片大小,并保存:https: // blog.csdn.net / SHAOYEZUIZUISHAUI / article / details / 103299503

上一篇:v4l2架构专题模块crop及selection分析 --- get_selection中V4L2_SEL_TGT_CROP_BOUNDS的目的


下一篇:真正的mybatiesPlus一键生成模板(根据对象增删改查分页的controller完善)详