Waifuset 是一个专门用于构建和管理文生图数据集的工具,其设计重点在于提供一个安全、高效、便捷、可以远程访问的用户界面(UI)。
- 安全:Waifuset 在操作过程中,不会修改数据集中的任何图像文件,确保数据的完整性和安全。
- 高效:大部分编辑操作在 Waifuset 中都非常高效,能够轻松处理十万乃至百万量级的大型数据集。
- 易用:UI 设计简洁直观,即使是复杂操作也可以轻松一键完成。
- 远程访问:通过 Web 服务进行访问,支持在本地或云服务器上部署和访问。
- UI 交互:Waifuset 的所有操作都通过用户界面完成,即使是没有编程背景的用户也能轻松上手。
- 基于标签的标注:Waifuset 假设所有的标注都是以英文逗号分隔的标签形式,而非自然语言描述。所有的标注编辑操作都将严格遵循这一格式。
- 专注于 Danbooru 标签:Waifuset 的标注优化算法主要面向 Danbooru 标签系统,不支持其他类型的标签体系。
- 标注专用:该工具主要用于编辑图像标注,而非图像本身。不包含图像编辑和文件系统功能。
- 数据集标注的构建和清洗:Waifuset 特别适用于构建和维护文生图数据集的标注,提供了一种高效和简便的方式。
- 数据集浏览:作为一个数据集浏览器,Waifuset 可以帮助用户有效地查看和管理他们的数据集。
为了使用 Waifuset,您需要首先安装项目及其依赖项。请按照以下步骤操作:
-
打开您的控制台。
-
输入以下命令以克隆项目并安装所需的依赖项:
git clone https://github.com/Eugeoter/waifuset cd waifuset pip install -r requirements.txt
-
如需使用 WaifuScorer 进行美学评分
如果您希望使用 WaifuScorer 进行美学评分,您需要安装额外的依赖项。在控制台中输入以下命令:
pip install torch torchvision pytorch-lightning huggingface-hub git+https://github.com/openai/CLIP.git
-
如需使用 WaifuTaggerV3 进行图像标注
若要使用 WaifuTaggerV3 对图像进行标注,您需要安装一些特定的依赖项。请在控制台中输入以下命令:
pip install torch torchvision onnxruntime huggingface-hub
- Windows: 点此查看 Windows 控制台打开方法
- MacOS: 点此查看 MacOS 控制台打开方法
- Linux: Linux 用户通常使用控制台进行操作。
Waifuset 的启动和使用非常直观。请遵循以下步骤快速开始:
-
启动 Waifuset
在控制台中,输入以下命令以启动 Waifuset:
python api.py --source path/to/folder --write_to_txt --language cn
path/to/folder
应该替换为您数据集的根目录路径。--write_to_txt
参数指示 Waifuset 将标注信息写入 txt 文件。--language cn
参数用于设置界面语言为中文。
-
访问 UI 界面
- 在命令成功执行后,控制台会输出一个本地 URL,例如
Running on local URL: http:https://xxx.xxx.xxx.xxx:xxxx
。 - 打开浏览器并输入显示的 URL(即
http:https://xxx.xxx.xxx.xxx:xxxx
部分)以访问 Waifuset 的用户界面。
- 在命令成功执行后,控制台会输出一个本地 URL,例如
-
云端服务器使用
- 如果您在云端服务器上部署 Waifuset,您可能需要启用
--share
参数以将 Gradio 网页共享到公网。 - 或者,您可以使用服务器的代理功能访问 UI 界面。
- 如果您在云端服务器上部署 Waifuset,您可能需要启用
按照这些步骤操作,您可以轻松地开始使用 Waifuset 来管理和编辑您的数据集。
为了更新 Waifuset,请遵循以下步骤:
-
打开您的控制台。
-
进入项目目录:
cd path/to/waifuset
注:请将
path/to/waifuset
替换为实际的项目路径。 -
执行以下命令以从远程仓库拉取最新的更新:
git pull
-
如果您之前修改过项目源码,例如更改过配置文件,这可能会导致冲突。
-
若要强制覆盖本地修改并更新项目,请使用:
git reset --hard git pull
-
若您希望通过 pip 安装 Waifuset,以便在您的 Python 环境中使用其类和函数,您可以直接从 GitHub 安装。请确保您的环境中已安装 git。然后,运行以下命令:
pip install git+https://github.com/Eugeoter/waifuset.git
或者,在控制台中打开 git clone 后的项目目录,然后运行以下命令:
pip install .
这将允许您在 Python 脚本中导入和使用 Waifuset。