您当前浏览的项目是 Waifuset1.0。目前,Waifuset 2.0 版本已经更新(点我浏览),其特点有:
通过在终端中运行以下命令更新至 Waifuset2.0 (需要 cd 到项目目录下):
git checkout v2
Waifuset 是一个专门用于构建和管理文生图数据集的工具,其设计重点在于提供一个安全、高效、便捷、可以远程访问的用户界面(UI)。
为了使用 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
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 界面
Running on local URL: http://xxx.xxx.xxx.xxx:xxxx
。http://xxx.xxx.xxx.xxx:xxxx
部分)以访问 Waifuset 的用户界面。云端服务器使用
--share
参数以将 Gradio 网页共享到公网。按照这些步骤操作,您可以轻松地开始使用 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。