制作了教师裁剪的拼图块后,发现如果图案上有黑色,就没法看清出凹凸角,剪起来有误差
所以黑色也要剔除
'''
通义万相下载的彩色蝴蝶有彩色背景的图片,放弃有白色和黑色图案的图片,保留彩色图案的蝴蝶
deepseek、阿夏
202505011'''import os
from PIL import Image# 定义源文件夹和目标文件夹
path=r'D:\彩色蝴蝶无白色3'# source_folder = path+r'\有颜色'
# target_folder = path+r'\有白色'
source_folder = path+r'\剔除黑白'
target_folder = path+r'\有黑白'# 确保目标文件夹存在
if not os.path.exists(target_folder):os.makedirs(target_folder)# 遍历源文件夹中的所有PNG文件
for filename in os.listdir(source_folder):if filename.lower().endswith('.png'):filepath = os.path.join(source_folder, filename)try:# 打开图片with Image.open(filepath) as img:# 转换为RGB模式(如果是RGBA或其他模式)rgb_img = img.convert('RGB')# 获取所有像素数据pixels = list(rgb_img.getdata())# 检查是否有白色像素(255, 255, 255)if (0,0,0) in pixels or (255, 255, 255) in pixels:# 构建目标路径target_path = os.path.join(target_folder, filename)# 移动文件os.rename(filepath, target_path)print(f"Moved {filename} to {target_folder}")except Exception as e:print(f"Error processing {filename}: {e}")print("Processing complete.")