szurubooru-toolkit
szurubooru-toolkit copied to clipboard
Deepbooru tag issue with missing rating due to threshold
I noticed that Deepbooru did not include a rating on a particular output, which also caused a tag to be missed as it was assumed by the toolkit to be a rating. I am using the latest 2022 release of the model and tags.txt from here: https://github.com/KichangKim/DeepDanbooru/releases/tag/v3-20211112-sgd-e28
Log file of szuru-toolkit:
[DEBUG] [07.09.2024, 01:20:09 EDT] [deepbooru.tag_image] Guessed following tags: ['1girl', 'blonde_hair', 'blue_background', 'blue_eyes', 'bodysuit', 'breasts', 'cleavage', 'gradient', 'gradient_background', 'large_breasts', 'lips', 'long_hair', 'no_bra', 'ponytail', 'scrunchie', 'signature', 'solo', 'zero_suit', 'samus_aran']
[DEBUG] [07.09.2024, 01:20:09 EDT] [utils.convert_rating] Converted rating samus_aran to None
[DEBUG] [07.09.2024, 01:20:09 EDT] [deepbooru.tag_image] Guessed rating None
When I ran deepdanbooru externally, it guessed the rating with a threshold lower than what I set with my toolkit config:
(0.603) rating:safe
I think that the rating should be assumed safe as a default if the guessed tags don't include a "rating:" item (or user configurable). This way the image would have still been tagged with "samus_aran"