Skip to content
Open
Show file tree
Hide file tree
Changes from 7 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion docs/wd14_tagger_README-en.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,11 @@ This document is based on the information from this github page (https://github.
Using onnx for inference is recommended. Please install onnx with the following command:

```powershell
pip install onnx==1.15.0 onnxruntime-gpu==1.17.1
pip install onnx onnxruntime-gpu
```

See [the official documentation](https://onnxruntime.ai/docs/install/#python-installs) for more details.

The model weights will be automatically downloaded from Hugging Face.

# Usage
Expand Down Expand Up @@ -49,6 +51,8 @@ python tag_images_by_wd14_tagger.py --onnx --repo_id SmilingWolf/wd-swinv2-tagge

# Options

All options can be checked with `python tag_images_by_wd14_tagger.py --help`.

## General Options

- `--onnx`: Use ONNX for inference. If not specified, TensorFlow will be used. If using TensorFlow, please install TensorFlow separately.
Expand Down
6 changes: 5 additions & 1 deletion docs/wd14_tagger_README-ja.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,11 @@
onnx を用いた推論を推奨します。以下のコマンドで onnx をインストールしてください。

```powershell
pip install onnx==1.15.0 onnxruntime-gpu==1.17.1
pip install onnx onnxruntime-gpu
```

詳細は[公式ドキュメント](https://onnxruntime.ai/docs/install/#python-installs)をご覧ください。

モデルの重みはHugging Faceから自動的にダウンロードしてきます。

# 使い方
Expand Down Expand Up @@ -48,6 +50,8 @@ python tag_images_by_wd14_tagger.py --onnx --repo_id SmilingWolf/wd-swinv2-tagge

# オプション

全てオプションは `python tag_images_by_wd14_tagger.py --help` で確認できます。

## 一般オプション

- `--onnx` : ONNX を使用して推論します。指定しない場合は TensorFlow を使用します。TensorFlow 使用時は別途 TensorFlow をインストールしてください。
Expand Down
Loading