制作了教师裁剪的拼图块后,发现如果图案上有黑色,就没法看清出凹凸角,剪起来有误差
所以黑色也要剔除
'''
通义万相下载的彩色蝴蝶有彩色背景的图片,放弃有白色和黑色图案的图片,保留彩色图案的蝴蝶
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.")