This project is dedicated to creating and running AI models that can automatically select appropriate tags for images and videos (including at specific timestamps). Having this tag data for images and videos can provide invaluable information to help manage content and find content without spending manual time hunting for the right videos or images.
The AI models available here are state-of-the-art, boasting far higher accuracy than any other NSFW AI models found on the internet. These models provide data that can be useful in many ways:
Precision Time-Based Tagging
Enhanced Searchability
Improved Content Management
Future Content Recommendations
The main difference between the free model and the patreon models is the number of supported tags. See the supported tags of the free and Patreon models below along with a rough plan for future tags that will be added in the future:
Before becoming a patron or installing the free model, please be aware of the following limitations:
Usage Restrictions: This code and associated models can only be used for personal local use as per the license. For any for-profit use cases or use cases that require online deployment, please contact me for licensing options.
Hardware Requirements: Only NVIDIA GPUs are supported. Any NVIDIA GPUs older than the NVIDIA 1080 will likely not work. Support for AMD GPUs is not planned.
Performance on CPU: While CPU is supported, it will be much slower compared to running on a GPU.
Complexity and Support: Running machine learning models is complex and requires precise conditions to work smoothly. Although I have worked to make the installation process and AI model predictions as user-friendly as possible, due to the variability in hardware and software environments, there might be issues. I will do my best to help resolve any issues, but I cannot guarantee that the models will run on every computer.
Model Access: The most capable models are for paid patrons only because this project takes an enormous amount of time, and Patreon funds future development. There is a free model available with less tags for users that may not care about all the tags or want to try the AI before considering becoming a patron.
This project includes both a free open source AI model as well as several paid Patreon only models. The free AI model supports 10 tags compared to the 36 currently supported tags in the Patreon variant. The paid version also includes different model variations that have different tradeoffs between accuracy and speed as well as more AI models for additional tag categories. You can see the list of currently available models here:
To see the benefits of NSFW AI Tagging for yourself, feel free to join as a Patron Member or try the free AI model:
If you have any questions, run into issues when installing, or have ideas for the future or just want to chat about the project, feel free to join my discord server and I'd be happy to chat!