Get a weekly rundown of the latest AI models and research... subscribe! https://aimodels.substack.com/

Cafe_aesthetic

cafeai

πŸ‘¨β€πŸ«

Info Since people are downloading this and I don't know why, I'll add some information. This model is an image classifier fine-tuned on microsoft/beit-base-patch16-384. Its purpose is to be used in the dataset conditioning step for the Waifu Diffusion project, a fine-tune effort for Stable Diffusion. As WD1.4 is planned to have a significantly large dataset (~15m images), it is infeasible to analyze every image manually to determine whether or not it should be included in the final training dataset. This image classifier is trained on approximately 3.5k real-life and anime/manga images. Its purpose is to remove aesthetically worthless images from our dataset by classifying them as "not_aesthetic". The image classifier was trained to err on the side of caution and will generally tend to include images unless they are in a "manga-like" format, have messy lines and/or are sketches, or include an unacceptable amount of text (namely text that covers the primary subject of the image). The idea is that certain images will hurt a SD fine-tune. Note: This classifier is not perfect, just like every other classifier out there. However, with a sufficiently large dataset, any imperfections or misclassifications should average themselves out due to the Law of Large Numbers. You can test out the classifier here, along with some other classifiers for the project. License Released under the aGPLv3. Use the model as you wish for any purpose. If you make changes, share the changes.
image-classification

Pricing

Cost per run
$-
USD
Avg run time
-
Seconds
Hardware
-
Prediction

Creator Models

ModelCostRuns
Cafe_style$?1,421
Cafe Instagram Sd 1 5 V6$?0
Cafe_waifu$?5,332
Wd14 Vae$?0

Similar Models

Try it!

You can use this area to play around with demo applications that incorporate the Cafe_aesthetic model. These demos are maintained and hosted externally by third-party creators. If you see an error, message me on Twitter.

Overview

Summary of this model and related resources.

PropertyValue
Creatorcafeai
Model NameCafe_aesthetic
Description

Info Since people are downloading this and I don't know why...

Read more Β»
Tagsimage-classification
Model LinkView on HuggingFace
API SpecView on HuggingFace
Github LinkNo Github link provided
Paper LinkNo paper link provided

Popularity

How popular is this model, by number of runs? How popular is the creator, by the sum of all their runs?

PropertyValue
Runs2,603
Model Rank
Creator Rank

Cost

How much does it cost to run this model? How long, on average, does it take to complete a run?

PropertyValue
Cost per Run$-
Prediction Hardware-
Average Completion Time-