主文件夹下多个子文件夹里面长方形图像按照长边批量填充为正方形,多余像素用其他色填充

 输入主文件夹目录:D:\flower\flower_photos

下面有多个子文件夹

import os
import glob
import cv2
import numpy as np

data_dir = 'D:/flower/flower_photos'#输入主文件夹路径
file = 'D:/flower/padding'#输出路径

for parent1, dirnames1, filenames1 in os.walk(data_dir):
    for dirname1 in dirnames1:
        rootdir = parent1 + '/' + dirname1
        for parent, dirnames, filenames in os.walk(rootdir):
            for filename in filenames:
                img = os.path.join(parent + '/' + filename)
                img = cv2.imread(img)
                size = img.shape
                h, w = size[0], size[1]
                m = max(w, h)
                resize_img = cv2.resize(img, (w, h))
                if m == w:
        
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值