Image Aspect Ratio Filter

Image Aspect Ratio Filter 图像长宽比过滤 Skill

功能概述

本skill根据图像的长宽比（宽/高）进行过滤，只保留长宽比在指定范围内的图像。 常用于筛选特定比例的图像，如正方形、横屏、竖屏等。

触发条件

当用户请求以下任务时，应使用此skill：

• 图像长宽比过滤
• 图片比例筛选
• 图像尺寸过滤
• 图像宽高比筛选
• 筛选正方形图像
• 筛选横屏/竖屏图像

核心参数说明

必需参数

可选参数

输入文件格式

[
  {"images": ["/path/to/image1.jpg"]},
  {"images": ["/path/to/image2.jpg"]},
  {"images": ["/path/to/image3.png"]}
]

长宽比说明

• 长宽比 = 宽 / 高
• 1.0 = 正方形
• < 1.0 = 竖屏图像

• 1.0 = 横屏图像

使用方法

默认参数过滤（长宽比0.333-3.0）

python scripts/run_image_aspect_ratio_filter.py \
  --input ./input.json \
  --output ./output.json

筛选正方形图像（长宽比接近1.0）

python scripts/run_image_aspect_ratio_filter.py \
  --input ./input.json \
  --output ./output.json \
  --min_ratio 0.9 \
  --max_ratio 1.1

筛选横屏图像

python scripts/run_image_aspect_ratio_filter.py \
  --input ./input.json \
  --output ./output.json \
  --min_ratio 1.0

筛选竖屏图像

python scripts/run_image_aspect_ratio_filter.py \
  --input ./input.json \
  --output ./output.json \
  --max_ratio 1.0

过滤策略说明

输出示例

命令行输出：

[OK] Image aspect ratio filtering completed!
   Min ratio: 0.333
   Max ratio: 3.0
   Strategy: any
   Original documents: 3
   Filtered documents: 2
   Removed documents: 1
   Input file: ./input.json
   Output file: ./output.json

输出JSON格式：

[
  {"images": ["/path/to/image1.jpg"]},
  {"images": ["/path/to/image2.jpg"]}
]

环境要求

安装依赖： 本SKILL使用依赖 data_juicer，请在调用前安装好python环境并安装data_juicer：

pip install py-data-juicer

注意事项

1. 图像路径：输入JSON中的图像路径应为绝对路径或相对于工作目录的路径
2. 长宽比计算：宽 / 高，1.0 表示正方形
3. 支持格式：支持 PNG、JPG、JPEG 等常见图像格式
4. 适用场景：筛选特定比例的图像、过滤极端比例图像