项目简介
本项目致力于利用DeepDanbooru模型对Photoprism中的图像开展标签提取与添加工作,达成动漫图像的多标签分类任务。通过整合DeepDanbooru和Photoprism,可自动将识别出的标签添加到Photoprism中,提升图像分类的准确性与效率。
项目的主要特性和功能
- 借助DeepDanbooru模型对图像进行多标签分类。
- 自动将识别出的标签添加到Photoprism数据库。
- 支持从文件夹批量处理图像。
- 提供命令行接口,便于用户进行批量操作。
安装使用步骤
环境准备
- 安装Python 3.7及以上版本。
- 安装TensorFlow库。
- 安装Photoprism Python SDK。
项目设置
修改config.py
文件,设置DeepDanbooru模型文件路径、标签文件路径、Photoprism服务器信息、索引路径等。
运行脚本
执行create_labels.py
脚本,开始处理图像并添加标签到Photoprism。
结果查看
可在Photoprism中查询添加的新标签。
注意事项
- 确保DeepDanbooru模型文件和标签文件路径正确。
- 确保Photoprism服务器信息正确,包括服务器地址、用户名和密码。
- 根据需要调整索引路径和批量处理设置。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】