Update README.md

README.md

@@ -7,4 +7,14 @@ metrics:
 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 |