Skip to content

Remake localtagger with imgutils for more models #2

@AshtakaOOf

Description

@AshtakaOOf

Localtagger is the backend of Farterrogator which is a material theme frontend for tagging images with danbooru tags.

Goal: More easily maintainable backend for adding new models etc with dghs-imgutils and the existing Kaloscope code for artist similarity.

dghs-imgutils is available as a pypi wheel https://pypi.org/project/dghs-imgutils/
this will be used to inference all the models we'll be adding as part of the remake (it can do dbv4 wd camie)

Models:

note1: animetimm's repos are locked behind agreeing to share email
note2: we cannot use onnx for animetimm as they are only packaged as safetensors as such use pytorch
note3: camie can guess artist but it's bad at it so omit the feature

Preferably the code should be well structured and not plopped into one file.
Also if possible easily configurable to choose where the models are hosted etc.

Goodies (optional): Send model list to frontend so backend doesn't have to updated to use newly added models. Ask developer how to tackle i18n if this is done.

Post remake: update frontend to the new api so the backend is actually usable from it.

Metadata

Metadata

Labels

enhancementNew feature or request

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions