|
楼主 |
发表于 2022-3-24 20:18:29
|
显示全部楼层
第三步,把第一张和最后两张图片移动到目标文件夹,剩余图片固定区域填充白色
- import numpy as np
- import os,shutil
- from PIL import Image
- def tianchong(url):
- #新建文件夹,用来存储处理完的图片
- if not os.path.exists(url + '-填充后'):
- os.mkdir(url + '-填充后')
- # 把不需要处理的三张图片,移动到目标文件夹
- files = os.listdir(url)
- changdu = len(files)
- shutil.move(f"{url}\\01.png", url + '-填充后')
- shutil.move(f'{url}\\{int(changdu)}.png', url + '-填充后')
- shutil.move(f'{url}\\{int(changdu)+ 1}.png', url + '-填充后')
- # 把所有文件遍历出来
- files = [url+'\\'+v for v in os.listdir(url)]
- # print(files)
- for a in files:
- newname = a.split('】')[0]+'】-填充后'+'\\'+a.split('\\')[-1]
- # 开始处理图片
- img = Image.open(a)
- array = np.array(img)
- rows = len(array)
- cols = len(array[0])
- # 图片尺寸
- # print(rows)
- # print(cols)
- # 设置图片修改后图片颜色 作者这里设置为白色
- colorl = [255,255,255]
- # 依次遍历我们需要修改颜色的图片区域
- for col in range(100, 870):
- for row in range(830, 1060):
- array[row, col] = colorl
- new_img = Image.fromarray(array)
- # new_img.show()
- new_img.save(newname)
- print(a, '已经处理完毕')
- path = input('请输入文件夹地址:')
- A = os.listdir(path)
- for a in A:
- tianchong(path+'\\'+a)
复制代码
|
|