Awesome

danbooru-tagger

Pytorch code for tagging dabooru images

Includes a pretrained model for tagging danbooru images. Trained on the Danbooru2019 512x512 SFW subset to predict the 6000 most common 'Category 0' tags. Achieves an F2 score of 0.61 on hold out test set, with a threshold of 7.9. For more performance information see the test_tagger.ipynb notebook.