1.4 KiB
1.4 KiB
图片彩色/黑白分类脚本
使用 Pillow 判定图片是否包含彩色信息，将文件按彩色或黑白/灰阶复制（或移动）到指定目录。默认输入目录为 data/color/mix，默认输出目录为当前目录下自动创建的 data/c-<timestamp> 与 data/b-<timestamp>。
环境准备
- Python 3.12+（建议与本机一致）
- 安装依赖：
pip install -r requirements.txt
基本用法
在仓库根目录运行（否则请用绝对路径或相对路径调用脚本）：
python classify_images.py \
--input-dir data/color/mix \
--color-dir data/c-$(date +%Y%m%d-%H%M%S) \
--bw-dir data/b-$(date +%Y%m%d-%H%M%S)
主要参数：
-i, --input-dir：待分类图片目录，默认data/color/mix。-c, --color-dir：彩色图片输出目录，默认data/c-<timestamp>。-b, --bw-dir：黑白/灰阶输出目录，默认data/b-<timestamp>。-t, --tolerance：颜色通道差异阈值，越大越宽松，默认 3。--move：移动文件（默认复制）。--dry-run：只显示分类结果与进度，不写入文件。
注意事项
- 默认路径为相对路径，请在仓库根目录运行，避免生成到意外位置。
- 处理进度每 10 张输出一次（最后一批也会输出）。
- 输入目录不存在或非目录时，脚本会打印用法并退出。