将图片补成正方形,不够用黑色补

import cv2
import os
import numpy as np
def get_resize_square(input,output):#,output
    for filename in os.listdir(input):
        image_path=input+'/'+filename
        image=cv2.imread(image_path)
        print(image)
        x=image.shape[1]##宽度
        y=image.shape[0]##高度
        print(x,y)###
        add_array=np.zeros((x-y, x))#宽度787,高度787-648=139
        #add_array_image = cv2.cvtColor(add_array, cv2.COLOR_GRAY2BGR)
        print('add',add_array)
        add_array = np.expand_dims(add_array, axis=2)
        image1 = np.concatenate((add_array, add_array, add_array), axis=-1)
        image_last = np.vstack((image, image1))
        print('image_last',image_last)
        print(image_last.shape)
        cv2.imwrite(output+'/'+filename,image_last)

if __name__=='__main__':
    input_image_dir=r'G:\shuju_quan\suoyou\3000'
    output_dir=r'G:\shuju_quan\suoyou\xiugai'
    get_resize_square(input_image_dir,output_dir)
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值