图片彩色/黑白分类脚本
使用 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 张输出一次(最后一批也会输出)。
- 输入目录不存在或非目录时,脚本会打印用法并退出。
Description
Languages
Python
100%