| --- |
| license: apache-2.0 |
| pipeline_tag: image-feature-extraction |
| new_version: nightknocker/contrastive-masked-pixiv-embedding |
| --- |
| |
| # Contrastive Booru-Image Embedding Model |
|
|
| The model was trained on pixiv images, tagged by [CL tagger](https://huggingface.co/cella110n/cl_tagger). |
| The caption (prompt) was tokenized by [ModernBERT](https://huggingface.co/answerdotai/ModernBERT-base). |
|
|
| ## Objectives |
|
|
| *Image-Caption Similarity Scoring*: The model assigns a score to each image based on how closely it is related to its associated caption. |
|
|
| *Image Labeling*: The image embedding is used to label images by color and character name. |
|
|
| ## Updates |
|
|
| The 1.2 and 1.3 versions focused on the heads, clothes, and colors of the most common anime characters. |
