改正代码:import os from PIL import Image import glob import numpy as np # 遍历文件夹 folder_path = 'E:/机器学习/helefull/labels' folders = os.listdir(folder_path) # print(folders) for filename in glob.glob(r'E:/机器学习/helefull/labels/*.png'): img=Image.open(filename).convert("RGB") # images=np.asarray(img) # print(images) # 只处理其中的20个文件夹 folder for folder in folders[:20]: folder_full_path = os.path.join(folder_path, folder) # print(folder_full_path) if os.path.isdir(folder_full_path): images = os.listdir(folder_full_path) print(images) blank_img = Image.new('RGB', (417, 354), (0, 0, 0)) for image_name in images: # 打开当前图片 img_path = os.path.join(folder_full_path, image_name) img = Image.open(img_path) # 遍历每一个像素点 for x in range(img.width): for y in range(img.height): # 如果当前像素点值为255,则将该像素点在空白图片上标记为i+1 if img.getpixel((x, y)) == 255: blank_img.putpixel((x, y), int(image_name.split('.')[0])+100) blank_img.save(f'new_{folder}.png')
时间: 2023-06-18 11:02:36 浏览: 42
import
from PIL import Image
import glob
import numpy as np
# 遍历文件夹
folder_path = 'E:/机器学习/helefull/labels'
folders = os.listdir(folder_path)
for filename in glob.glob(r'E:/机器学习/helefull/labels/*.png'):
img = Image.open(filename).convert("RGB")
# images=np.asarray(img)
# print(images)
# 只处理其中的20个文件夹
for folder in folders[:20]:
folder_full_path = os.path.join(folder_path, folder)
# print(folder_full_path)
if os.path.isdir(folder_full_path):
images = os.listdir(folder_full_path)
print(images)
# 创建一个空白图片
blank_img = Image.new('RGB', (417, 354), (0, 0, 0))
for image_name in images:
# 打开当前图片
img_path = os.path.join(folder_full_path, image_name)
img = Image.open(img_path)
# 遍历每一个像素点
for x in range(img.width):
for y in range(img.height):
# 如果当前像素点值为255,则将该像素点在空白图片上标记为i+1
if img.getpixel((x, y)) == (255, 255, 255):
blank_img.putpixel((x, y), int(image_name.split('.')[0])+100)
blank_img.save(f'new_{folder}.png')