stashapp / CommunityScripts

This is a public repository containing plugin and utility scripts created by the Stash Community.
https://docs.stashapp.cc/add-ons/
GNU Affero General Public License v3.0
180 stars 140 forks source link

[Feature] Tagging Images using Neural Net #415

Open eloiselle opened 1 year ago

eloiselle commented 1 year ago

Is your feature request related to a problem? Please describe. I've been going ham trying to fill my stash app with a bunch of pictures, but since I've been downloading from multiple sources, it's very hard to track and tag every picture with it's contents. If you know what kind of content you're feeling like "watching", this is a concise way to organize your library.

Describe the solution you'd like I've been doing some research, and I know it's a big endeavor, but amongst all the NSFW classifier projects, this one stood out: NudeNet

A big feature of this one seems to be its ability to classify pictures into a bunch of categories, for example:

class name Description
EXPOSED_BELLY Exposed Belly; Any gender
EXPOSED_BUTTOCKS Exposed Buttocks; Any gender
EXPOSED_BREAST_F Exposed Breast; Female
EXPOSED_GENITALIA_F Exposed Genitalia; Female
EXPOSED_GENITALIA_M Exposed Genitalia; Male
EXPOSED_BREAST_M Exposed Breast; Male

Describe alternatives you've considered I've been trying to get this one to work so it could organize my pictures into folders with the above classes, to no great success so far. It would probably be less complicated to setup, but I thought I should bring this up.

I've also taken a look at a bunch of projects incorporating NudeNet's code (see this search result), but none of them seem to do quite the task I want them to do.

Additional context I've been looking at this issue for quite a while now, yet there hasn't been much work put into such a feature. Most NSFW classifiers out there are specifically to identify if something is SFW / NSFW, instead of organizing WHY they're NSFW.

If anyone has ideas about how this problem could be solved, feel free to let me know. I mostly want to open a conversation about this topic.

[Edit] Seeing other feature requests, I feel like this would probably be a plugin of some sort?

stg-annon commented 1 year ago

There is an effort to do things like this on the Discord look for the channel #ai-recog-tagging however I do not see this being built into Stash directly but rather as a Plugin as a couple of tools currently are, see StashTag as an example