kapitan/wdtagger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(dependencies): add image tagging script && update torch dependency

Browse Source
This commit is contained in:
Jianqi Pan 2025-02-21 15:06:01 +09:00
parent 92ab8bcbcc
commit 458a0d9410
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
a.py Normal file
View File

@ -0,0 +1,10 @@
from PIL import Image
from wdtagger import Tagger
if __name__ == "__main__":
tagger = Tagger()
images = [
Image.open("./tests/images/赤松楓.9d64b955.jpeg"),
]
results = tagger.tag(images)

4
pyproject.toml
View File

@ -17,8 +17,8 @@ dev = ["pytest>=8.3.3", "pytest-benchmark>=5.1.0", "ruff>=0.8.0"]
[project.optional-dependencies]
cpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
gpu = ["torch>=2.5.1", "torchvision>=0.20.1", "timm>=1.0.11"]
cpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
gpu = ["torch>=2.0.0", "torchvision>=0.20.1", "timm>=1.0.11"]
[tool.ruff]
line-length = 140