diff --git a/README.md b/README.md index aacc83d..8771e1c 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,8 @@ A self-hosted image and video gallery application designed for organizing and vi - **Video Support** - Playback for video files with automatic transcoding to MP4 - **Tagging System** - Organize images with tags (artist, character, series, rating, archive, user-defined) - **Tag Autocomplete** - Quick tag entry with search suggestions -- **ML Tag Suggestions** - WD14 tagger + SigLIP embedding centroids propose tags you can accept/reject per image +- **ML Tag Suggestions** - WD14 tagger + SigLIP embedding centroids propose tags you can accept/reject per image. Videos are sampled at 10 frames (configurable via `VIDEO_ML_FRAMES`) and aggregated by max-confidence WD14 + mean-pool SigLIP. +- **Bulk Tag Suggestions** - Select multiple images in the gallery to see consensus ML suggestions (tags suggested for or already applied to >=80% of the selection) - **Automatic Importing** - Celery-based task queue scans `/import` directory on schedule - **Archive Extraction** - Supports ZIP, RAR, 7z and other archive formats - **Duplicate Detection** - Perceptual hash (pHash) comparison to skip similar images @@ -196,6 +197,8 @@ These can also be configured via the Settings page in the UI. | `SIGLIP_REPO` | `google/siglip-so400m-patch14-384` | HuggingFace repo for the SigLIP encoder | | `WD14_REVISION` | (latest) | Pin WD14 to a specific commit SHA | | `SIGLIP_REVISION` | (latest) | Pin SigLIP to a specific commit SHA | +| `VIDEO_ML_FRAMES` | `10` | Frames sampled per video for WD14+SigLIP inference (between 10% and 90% of duration) | +| `WD14_STORE_FLOOR` | `0.05` | Minimum WD14 confidence stored in the DB | ### Other