Update README.md
Browse files
README.md
CHANGED
@@ -25,14 +25,16 @@ The model used to predict the types of anime images, which includes the followin
|
|
25 |
| mobilenetv3_sce_dist | 0.63G | 92.35% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_sce_dist/plot_confusion.png) | Distrillated from caformer_s36_plus, using mobilenetv3_large_100 with SCELoss |
|
26 |
| mobilevitv2_150 | 9.09G | 88.21% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilevitv2_150/plot_confusion.png) | Model: mobilevitv2_150 from timm |
|
27 |
|
28 |
-
|
|
29 |
-
|
30 |
-
|
|
31 |
-
|
|
32 |
-
|
|
33 |
-
|
|
34 |
-
|
|
35 |
-
|
|
36 |
-
|
|
37 |
-
|
|
38 |
-
|
|
|
|
|
|
|
25 |
| mobilenetv3_sce_dist | 0.63G | 92.35% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_sce_dist/plot_confusion.png) | Distrillated from caformer_s36_plus, using mobilenetv3_large_100 with SCELoss |
|
26 |
| mobilevitv2_150 | 9.09G | 88.21% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilevitv2_150/plot_confusion.png) | Model: mobilevitv2_150 from timm |
|
27 |
|
28 |
+
| Name | FLOPS | Params | Accuracy | AUC | Confusion | Labels |
|
29 |
+
|:-----------------------:|:-------:|:--------:|:----------:|:------:|:---------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------:|
|
30 |
+
| caformer_s36 | 22.10G | 37.22M | 88.19% | N/A | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
31 |
+
| caformer_s36_plus | 22.10G | 37.22M | 93.47% | 0.9891 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_plus/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
32 |
+
| caformer_s36_v1.1_focal | 22.10G | 37.22M | 95.99% | 0.9967 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.1_focal/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
|
33 |
+
| caformer_s36_v1.2_focal | 22.10G | 37.22M | 97.23% | 0.9982 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.2_focal/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
|
34 |
+
| caformer_s36_v1 | 22.10G | 37.22M | 94.72% | 0.9934 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
|
35 |
+
| mobilenetv3 | 0.63G | 4.18M | 88.96% | N/A | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
36 |
+
| mobilenetv3_dist | 0.63G | 4.18M | 91.98% | 0.9879 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_dist/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
37 |
+
| mobilenetv3_sce | 0.63G | 4.18M | 89.92% | 0.9786 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_sce/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
38 |
+
| mobilenetv3_sce_dist | 0.63G | 4.18M | 92.35% | 0.9854 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_sce_dist/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|
39 |
+
| mobilenetv3_v1_dist | 0.63G | 4.18M | 94.04% | 0.9928 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_v1_dist/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
|
40 |
+
| mobilevitv2_150 | 9.09G | 9.79M | 88.21% | N/A | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilevitv2_150/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration` |
|