search

TagStudioDev / TagStudio

A User-Focused Photo & File Management System
https://docs.tagstud.io/
GNU General Public License v3.0
5.18k stars 371 forks source link

[Feature Request]: Ai Tagging #326

Open ThesiiNCey opened 3 months ago

ThesiiNCey commented 3 months ago

Checklist

Description

I have write an Ai Tagging Script for my Local Images in Python thats using https://huggingface.co/SmilingWolf/wd-v1-4-swinv2-tagger-v2

Can you build a Feature Like this in ?

Or can it help to share my Script ?

Solution

Add an Image Interogator to Tag Images Automaticly Like in Stable Diffusion ( WD Tagger )

Alternatives

No response

Qronikarz commented 2 months ago

Is this issue about general AI tagging? Or just tagging from one source (like Stable Diffusion tags)?

If the first case then it could be implemented like in Spacedrive: https://www.spacedrive.com/blog/alpha-zero-two-release It used AI to automatically tag images using one of the models. Though the tags that were created this way had a different name so that it is known that they were applied automatically.

This model would be especially useful: https://github.com/xinyu1205/recognize-anything Though I have no idea how hard it would be to implement it, or what does it need and how much space it would take.