nightknocker's picture
Update README.md
5df3ced verified
|
raw
history blame
744 Bytes
metadata
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. The caption (prompt) was tokenized by ModernBERT.

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.