Anime Rating

--- license: mit datasets: - deepghs/anime_rating metrics: - accuracy pipeline_tag: image-classification tags: - art --- Image rating (`safe`, `r15`, `r18`) prediction model. Please note that due to the often unclear boundaries between `safe`, `r15` and `r18` levels, there is no objective ground truth for this task, and the data is scraped without any manual filtering. Therefore, the models trained on this dataset can only provide rough judgments. If you require an accurate solution for classifying `R18` images, it is recommended to consider a solution based on keypoint object detection. | Model | FLOPs | Accuracy | Confusion Matrix | Description | |:--------------------:|:------:|:--------:|:-----------------------------------------------------------------------------------------------------------------:|----------------------------------------------------------------------| | caformer_s36_plus | 22.10G | 74.26% | [Confusion Matrix](https://huggingface.co/deepghs/anime_rating/blob/main/caformer_s36_plus/plot_confusion.png) | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm | | mobilenetv3 | 0.63G | 64.77% | [Confusion Matrix](https://huggingface.co/deepghs/anime_rating/blob/main/mobilenetv3/plot_confusion.png) | Model: mobilenetv3_large_100 from timm | | mobilenetv3_sce | 0.63G | 66.27% | [Confusion Matrix](https://huggingface.co/deepghs/anime_rating/blob/main/mobilenetv3_sce/plot_confusion.png) | Model: mobilenetv3_large_100 from timm, use SCELoss as loss function | | mobilenetv3_sce_dist | 0.63G | 69.49% | [Confusion Matrix](https://huggingface.co/deepghs/anime_rating/blob/main/mobilenetv3_sce_dist/plot_confusion.png) | Distillated from caformer_s36_plus, using mobilenetv3_large_100 |

Source context: 0 downloads · 0 likes · Pipeline image-classification · Repo Library-Mutsumi/anime_rating