KichangKim/DeepDanbooru

Features

• Anime Image Taggers - Analyzes anime-style images and outputs Danbooru-style tags identifying characters and attributes.
• Custom Model Training - Trains custom tag recognition models using SQLite datasets and configurable hyperparameters.
• Deep Learning Inference Engines - Runs a pre-trained convolutional neural network to predict tags from image pixel data.
• Model-Based Tag Predictions - Evaluates images against a trained model to produce predicted tags for each image.
• Image Tag Training Pipelines - Processes user-provided images and tags into a training workflow with hyperparameter optimization.
• Training Dataset Preparation - Organizes images and tags into a structured SQLite database for the training pipeline.
• Remote Data Fetching - Fetches tag definitions and metadata from a remote server using authenticated API requests.
• SQLite Storage Adapters - Stores image-tag pairs in a structured SQLite database for efficient training data management.
• Tag Definition Downloads - Downloads Danbooru tag definitions via authenticated API for training and estimation tasks.
• Rating and Score Filters - Filters raw SQLite datasets by rating and score criteria before training.

Star history