Compare commits
2 Commits
Author | SHA1 | Date | |
---|---|---|---|
8c8ca21b3d | |||
094960e721 |
@ -1,6 +1,6 @@
|
||||
[project]
|
||||
name = "wdtagger"
|
||||
version = "0.13.2"
|
||||
version = "0.14.0"
|
||||
description = "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."
|
||||
authors = [{ name = "Jianqi Pan", email = "jannchie@gmail.com" }]
|
||||
readme = "README.md"
|
||||
|
@ -125,7 +125,7 @@ class Result:
|
||||
"""Return general tags as a tuple."""
|
||||
|
||||
return tuple(
|
||||
d.replace("_", " ").replace("(", R"\(").replace(")", R"\)")
|
||||
d.replace("_", " ")
|
||||
for d in sorted(
|
||||
self.general_tag_data,
|
||||
key=lambda k: self.general_tag_data[k],
|
||||
@ -137,7 +137,7 @@ class Result:
|
||||
def character_tags(self) -> tuple[str, ...]:
|
||||
"""Return character tags as a tuple."""
|
||||
return tuple(
|
||||
d.replace("_", " ").replace("(", R"\(").replace(")", R"\)")
|
||||
d.replace("_", " ")
|
||||
for d in sorted(
|
||||
self.character_tag_data,
|
||||
key=lambda k: self.character_tag_data[k],
|
||||
|
7
temp.py
Normal file
7
temp.py
Normal file
@ -0,0 +1,7 @@
|
||||
from wdtagger import Tagger
|
||||
|
||||
tagger = Tagger()
|
||||
|
||||
|
||||
resp = tagger.tag("./tests/images/GkZFQ-_WwAAeNlO.jpg", character_threshold=0.85, general_threshold=0.35)
|
||||
print(resp.character_tags)
|
BIN
tests/images/GkZFQ-_WwAAeNlO.jpg
Normal file
BIN
tests/images/GkZFQ-_WwAAeNlO.jpg
Normal file
Binary file not shown.
After Width: | Height: | Size: 706 KiB |
Reference in New Issue
Block a user