48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
|
import os
|
|||
|
|
import shutil
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# ================== 配置 ==================
|
|||
|
|
SOURCE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/class/class0/3" # 原始图片文件夹
|
|||
|
|
OUTPUT_A = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/class/class0/2" # a 子文件夹
|
|||
|
|
OUTPUT_B = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/class/class0/4" # b 子文件夹
|
|||
|
|
|
|||
|
|
# 支持的图片扩展名
|
|||
|
|
IMG_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp'}
|
|||
|
|
|
|||
|
|
# ================== 创建输出目录 ==================
|
|||
|
|
os.makedirs(OUTPUT_A, exist_ok=True)
|
|||
|
|
os.makedirs(OUTPUT_B, exist_ok=True)
|
|||
|
|
|
|||
|
|
# ================== 获取并排序图片 ==================
|
|||
|
|
source_path = Path(SOURCE_DIR)
|
|||
|
|
if not source_path.exists():
|
|||
|
|
raise FileNotFoundError(f"源文件夹不存在: {SOURCE_DIR}")
|
|||
|
|
|
|||
|
|
# 获取所有图片文件，并按名称排序（确保顺序可重现）
|
|||
|
|
image_files = sorted([
|
|||
|
|
f for f in source_path.iterdir()
|
|||
|
|
if f.is_file() and f.suffix.lower() in IMG_EXTENSIONS
|
|||
|
|
])
|
|||
|
|
|
|||
|
|
if not image_files:
|
|||
|
|
print("⚠️ 源文件夹中没有找到图片！")
|
|||
|
|
exit()
|
|||
|
|
|
|||
|
|
print(f"共找到 {len(image_files)} 张图片，开始分配...")
|
|||
|
|
|
|||
|
|
# ================== 交替分配到 a 和 b ==================
|
|||
|
|
for idx, img_path in enumerate(image_files):
|
|||
|
|
if idx % 2 == 0:
|
|||
|
|
# 偶数索引（0, 2, 4...）→ a
|
|||
|
|
dest = os.path.join(OUTPUT_A, img_path.name)
|
|||
|
|
else:
|
|||
|
|
# 奇数索引（1, 3, 5...）→ b
|
|||
|
|
dest = os.path.join(OUTPUT_B, img_path.name)
|
|||
|
|
|
|||
|
|
shutil.move(img_path, dest) # 移动原来数据
|
|||
|
|
print(f"[{idx + 1}/{len(image_files)}] {img_path.name} → {'a' if idx % 2 == 0 else 'b'}")
|
|||
|
|
|
|||
|
|
print("\n✅ 分配完成！")
|
|||
|
|
print(f"a 文件夹: {OUTPUT_A}")
|
|||
|
|
print(f"b 文件夹: {OUTPUT_B}")
|