2025-08-13 14:49:06 +08:00
|
|
|
|
import os
|
|
|
|
|
|
from PIL import Image
|
|
|
|
|
|
import numpy as np
|
|
|
|
|
|
|
|
|
|
|
|
def is_large_gray(image, gray_lower_threshold=70, gray_upper_threshold=230, gray_ratio_threshold=0.7):
|
|
|
|
|
|
"""
|
|
|
|
|
|
判断图片是否大面积为灰色（基于像素的颜色值）
|
|
|
|
|
|
|
|
|
|
|
|
参数：
|
|
|
|
|
|
- image: 图片对象（PIL Image）
|
|
|
|
|
|
- gray_lower_threshold: 灰色下限（低于此值不算“灰色”）
|
|
|
|
|
|
- gray_upper_threshold: 灰色上限（高于此值不算“灰色”）
|
|
|
|
|
|
- gray_ratio_threshold: 灰色像素占比阈值（>70% 算大面积灰色）
|
|
|
|
|
|
|
|
|
|
|
|
返回：True 表示是大面积灰色，应删除
|
|
|
|
|
|
"""
|
|
|
|
|
|
# 将图片转换为 numpy 数组
|
|
|
|
|
|
img_array = np.array(image)
|
|
|
|
|
|
|
|
|
|
|
|
# 获取图片的尺寸
|
|
|
|
|
|
height, width, _ = img_array.shape
|
|
|
|
|
|
total_pixels = height * width
|
|
|
|
|
|
|
|
|
|
|
|
# 判断是否为灰色像素（R、G、B 值都在 gray_lower_threshold 和 gray_upper_threshold 之间）
|
|
|
|
|
|
gray_pixels = np.sum(
|
|
|
|
|
|
(img_array[:, :, 0] >= gray_lower_threshold) &
|
|
|
|
|
|
(img_array[:, :, 0] <= gray_upper_threshold) &
|
|
|
|
|
|
(img_array[:, :, 1] >= gray_lower_threshold) &
|
|
|
|
|
|
(img_array[:, :, 1] <= gray_upper_threshold) &
|
|
|
|
|
|
(img_array[:, :, 2] >= gray_lower_threshold) &
|
|
|
|
|
|
(img_array[:, :, 2] <= gray_upper_threshold)
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
gray_ratio = gray_pixels / total_pixels
|
|
|
|
|
|
|
|
|
|
|
|
return gray_ratio > gray_ratio_threshold
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def process_images_in_folder(input_folder, output_folder):
|
|
|
|
|
|
"""
|
|
|
|
|
|
遍历文件夹，旋转图片并根据条件保存到输出文件夹
|
|
|
|
|
|
"""
|
|
|
|
|
|
# 创建输出文件夹（如果不存在）
|
|
|
|
|
|
if not os.path.exists(output_folder):
|
|
|
|
|
|
os.makedirs(output_folder)
|
|
|
|
|
|
|
|
|
|
|
|
# 支持的图片格式
|
|
|
|
|
|
supported_formats = ('.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.tif', '.webp')
|
|
|
|
|
|
|
|
|
|
|
|
for filename in os.listdir(input_folder):
|
|
|
|
|
|
file_path = os.path.join(input_folder, filename)
|
|
|
|
|
|
|
|
|
|
|
|
if not os.path.isfile(file_path):
|
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
|
|
if not filename.lower().endswith(supported_formats):
|
|
|
|
|
|
continue
|
|
|
|
|
|
|
|
|
|
|
|
print(f"处理: {filename}")
|
|
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
|
with Image.open(file_path) as img:
|
|
|
|
|
|
# 判断是否为大面积灰色
|
|
|
|
|
|
if is_large_gray(img):
|
|
|
|
|
|
print(f" 🔴 不保存大面积灰色图片: {filename}")
|
|
|
|
|
|
continue # 不保存该图片
|
|
|
|
|
|
|
|
|
|
|
|
# 否则：打开并旋转 180 度
|
|
|
|
|
|
rotated_img = img.rotate(180, expand=False)
|
|
|
|
|
|
|
|
|
|
|
|
# 构建新的保存路径
|
|
|
|
|
|
save_path = os.path.join(output_folder, filename)
|
|
|
|
|
|
|
|
|
|
|
|
# 保持原格式保存（覆盖原图）
|
|
|
|
|
|
rotated_img.save(save_path, format=img.format)
|
|
|
|
|
|
print(f" ✅ 已旋转并保存至: {save_path}")
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
print(f" ❌ 处理失败 {filename}: {e}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ================ 使用示例 ================
|
|
|
|
|
|
if __name__ == "__main__":
|
2025-08-14 18:24:45 +08:00
|
|
|
|
folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10"
|
|
|
|
|
|
output_folder = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f10"
|
2025-08-13 14:49:06 +08:00
|
|
|
|
|
|
|
|
|
|
if not os.path.exists(folder):
|
|
|
|
|
|
print("❌ 输入文件夹不存在！")
|
|
|
|
|
|
else:
|
|
|
|
|
|
process_images_in_folder(folder, output_folder)
|
|
|
|
|
|
print("✅ 所有图片处理完成！")
|
