70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
import os
|
||
import shutil
|
||
from pathlib import Path
|
||
|
||
def split_images_into_five(source_dir, target_root, move_instead_of_copy=True):
|
||
"""
|
||
将 source_dir 中的图片均分为 5 份，保存到 target_root/1 ~ 5/
|
||
|
||
:param source_dir: 源图片文件夹路径
|
||
:param target_root: 目标根目录（会创建 1~5 子文件夹）
|
||
:param move_instead_of_copy: True 表示移动，False 表示复制（默认为 True 移动）
|
||
"""
|
||
source = Path(source_dir)
|
||
target_root = Path(target_root)
|
||
|
||
# 支持的图片扩展名
|
||
image_extensions = {'.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.tif', '.webp'}
|
||
|
||
# 获取所有图片文件（按名称排序，保证可重复性）
|
||
all_images = sorted([
|
||
f for f in source.iterdir()
|
||
if f.is_file() and f.suffix.lower() in image_extensions
|
||
])
|
||
|
||
if not all_images:
|
||
print("❌ 源文件夹中没有找到图片！")
|
||
return
|
||
|
||
total = len(all_images)
|
||
print(f"📁 共找到 {total} 张图片")
|
||
|
||
# 均分为 5 份
|
||
chunk_size = (total + 4) // 5 # 向上取整，确保覆盖所有图片
|
||
|
||
# 创建 1~5 文件夹并分发图片
|
||
for i in range(5):
|
||
start_idx = i * chunk_size
|
||
end_idx = min((i + 1) * chunk_size, total)
|
||
if start_idx >= total:
|
||
break # 防止空分片
|
||
|
||
folder_name = str(i + 1)
|
||
target_folder = target_root / folder_name
|
||
target_folder.mkdir(parents=True, exist_ok=True)
|
||
|
||
chunk_files = all_images[start_idx:end_idx]
|
||
print(f"📦 文件夹 {folder_name}: 分配 {len(chunk_files)} 张图片")
|
||
|
||
for img_path in chunk_files:
|
||
dst = target_folder / img_path.name
|
||
if move_instead_of_copy:
|
||
shutil.move(str(img_path), str(dst))
|
||
print(f" 📂 已移动: {img_path.name}")
|
||
else:
|
||
shutil.copy2(str(img_path), str(dst))
|
||
print(f" 📂 已复制: {img_path.name}")
|
||
|
||
print(f"\n✅ 分割完成！结果保存在: {target_root}")
|
||
|
||
if __name__ == "__main__":
|
||
# ========================
|
||
# ⚙️ 配置你的路径
|
||
# ========================
|
||
SOURCE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/分割/class3"
|
||
TARGET_ROOT = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/分割/class3"
|
||
|
||
# 设置为 True 则移动图片（原图会被移走），False 则复制（推荐先用 False 测试）
|
||
MOVE_MODE = True
|
||
|
||
split_images_into_five(SOURCE_DIR, TARGET_ROOT, move_instead_of_copy=MOVE_MODE) |