使用场景

开发者使用该工具为网站图片批量生成描述性标题，提高搜索引擎优化(SEO)效果。

数据科学家利用该工具为数据集中的图片生成描述，用于机器学习模型的训练。

爱好者使用该工具为个人图片收藏生成有趣的描述，增加图片的趣味性。

产品特色

批量处理图片文件，生成描述性标题

支持自定义图片目录，不仅限于默认的./input目录

支持LOW_VRAM_MODE，适应显存较低的设备

提供详细的安装和使用说明，方便用户快速上手

支持Python 3.9至3.11版本，不包括3.12

需要安装PyTorch与CUDA支持，与CUDA版本相匹配

在GitHub上开源，用户可以自由下载和修改

使用教程

1. 使用git clone命令克隆该仓库到本地。

2. （可选）创建一个虚拟环境，并激活它。

3. 运行pip install -r requirements.txt安装所需的依赖。

4. 安装与CUDA版本相匹配的PyTorch。

5. 将需要生成描述的图片放入/input目录，或使用--img_dir参数指定其他目录。

6. 运行python batch.py命令，工具将自动为目录中的图片生成描述性标题。