python实现图片按照长短边比例进行缩放(不改变图片比例)

        cv中,图片预处理会用到修改图片的尺寸,直接resize()会改变图像的长宽比例,也就是变形,会影响后期的模型学习,这篇文章将介绍一种按照长短边比例缩放改变图片的大小。

        先上代码:

#!/usr/bin/env python 
# -*- coding:utf-8 -*-
# @Time : 2022/8/11 15:13
# @Author : coco
# @File : resize.py
#实现图像长短边等比例缩放。并补白色

def img_pad(pil_file):
    #h,w 先后不要写错,不然图片会变形
    h, w, c = pil_file.shape
    # print(h, w, c)
    fixed_size = 256  # 输出正方形图片的尺寸

    if h >= w:
        factor = h / float(fixed_size)
        new_w = int(w / factor)
        if new_w % 2 != 0:
            new_w -= 1
        pil_file = cv2.resize(pil_file, (new_w, fixed_size))
        pad_w = int((fixed_size - new_w) / 2)
        array_file = np.array(pil_file)
        # array_file = np.pad(array_file, ((0, 0), (pad_w, fixed_size-pad_w)), 'constant') #实现黑白图缩放
        array_file = cv2.copyMakeBorder(array_file, 0,0,pad_w,fixed_size - new_w - pad_w,cv2.BORDER_CONSTANT, value=(255, 255, 255))  #255是白色,0是黑色
    else:
        factor = w / float(fixed_size)
        new_h = int(h / factor)
        if new_h % 2 != 0:
            new_h -= 1
        pil_file = cv2.resize(pil_file, (fixed_size, new_h))
        pad_h = int((fixed_size - new_h) / 2)
        array_file = np.array(pil_file)
        # array_file = np.pad(array_file, ((pad_h, fixed_size-pad_h), (0, 0)), 'constant')
        array_file = cv2.copyMakeBorder(array_file, pad_h, fixed_size - new_h - pad_h, 0, 0, cv2.BORDER_CONSTANT, value=(255, 255, 255))
    output_file = Image.fromarray(array_file)
    return output_file


if __name__ == "__main__":
    dir_image = 'tianchi_sum'  # 图片所在文件夹
    dir_output = 'tianchi_sum1'  # 输出结果文件夹
    if not os.path.exists(dir_output):
        os.makedirs(dir_output)
    i = 0
    list_image = os.listdir(dir_image)
    for file in list_image:
        path_image = os.path.join(dir_image, file)
        path_output = os.path.join(dir_output, file)
        pil_image = cv2.imread(path_image)
        b, g, r = cv2.split(pil_image)  # 通道分离,再重新合并操作
        pil_image = cv2.merge([r, g, b])
        # print(pil_image)
        # pil_image = pil_image.load()
        output_image = img_pad(pil_image)
        output_image.save(path_output)
        i += 1
        if i % 1000 == 0:
            print('The num of processed images:', i)    #

图片效果展示:

        因为padding为255白色不太容易看出来,哈哈哈 可以选中图片,不改变长短比例进行缩放。如果需要的话还可以改变padding的颜色。

        就连太阳光到地球都需要8分钟,所以你也需要时间,耐心一些。

——董宇辉

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值