🎉 init(wdtagger): a simple and easy-to-use wrapper for wdtagger

This commit is contained in:
Jianqi Pan
2024-06-06 23:53:55 +09:00
commit 2b67275105
7 changed files with 1264 additions and 0 deletions

25
README.md Normal file
View File

@ -0,0 +1,25 @@
# wdtagger
`wdtagger` is a simple and easy-to-use wrapper for the tagger model created by [SmilingWolf](https://github.com/SmilingWolf) which is specifically designed for tagging anime illustrations.
## Installation
You can install `wdtagger` via pip:
```bash
pip install wdtagger
```
## Usage
Below is a basic example of how to use wdtagger in your project:
```python
from PIL import Image
from wdtagger import Tagger
tagger = Tagger() # You can provide the model_repo, the default is "SmilingWolf/wd-swinv2-tagger-v3"
image = Image.open("image.jpg")
result = tagger.tag(image)
print(result)
```