deepghs/wd14_tagger_inversion
What's this This is the dataset for inversing the embeddings of the given prediction result. We can get the embedding dataset with some tag combinations, and search images like this. How is this dataset made This dataset is generated with anime images in danbooru (webp ones, based on KBlueLeaf/danbooru2023-webp-4Mpixel), extracting the prediction results and the embeddings of each images. Model's goal Train a model to inverse prediction result to… See the full description on the dataset page: https://huggingface.co/datasets/deepghs/wd14_tagger_inversion.
What's this
This is the dataset for inversing the embeddings of the given prediction result. We can get the embedding dataset with some tag combinations, and search images like this.
How is this dataset made
This dataset is generated with anime images in danbooru (webp ones, based on KBlueLeaf/danbooru2023-webp-4Mpixel), extracting the prediction results and the embeddings of each images.
Model's goal
Train a model to inverse prediction result to embeddings. Maybe a MLP can be used? I don't know, just try them.