Update README.md
Browse files
README.md
CHANGED
@@ -15,10 +15,12 @@ The model used to predict the types of anime images, which includes the followin
|
|
15 |
* Comic: Images of manga that contain a significant amount of text or panel sequences.
|
16 |
* Illustration: General anime illustrations.
|
17 |
|
18 |
-
|
|
19 |
-
|
20 |
-
|
|
21 |
-
|
|
22 |
-
|
|
23 |
-
|
|
24 |
-
|
|
|
|
|
|
|
15 |
* Comic: Images of manga that contain a significant amount of text or panel sequences.
|
16 |
* Illustration: General anime illustrations.
|
17 |
|
18 |
+
| Model | FLOPs | Accuracy | Confusion Matrix | Description |
|
19 |
+
|:--------------------:|:------:|:--------:|:-------------------------------------------------------------------------------------------------------------------------:|----------------------------------------------------------------------------------|
|
20 |
+
| caformer_s36 | 22.10G | 88.19% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36/plot_confusion.png) | Model: caformer_s36 from timm |
|
21 |
+
| caformer_s36_plus | 22.10G | 93.47% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_plus/plot_confusion.png) | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm |
|
22 |
+
| mobilenetv3 | 0.63G | 88.96% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3/plot_confusion.png) | Model: mobilenetv3_large_100 from timm |
|
23 |
+
| mobilenetv3_dist | 0.63G | 91.98% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_dist/plot_confusion.png) | Distrillated from caformer_s36_plus, using mobilenetv3_large_100 with focal loss |
|
24 |
+
| mobilenetv3_sce | 0.63G | 89.92% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_sce/plot_confusion.png) | Model: mobilenetv3_large_100 from timm, use SCELoss as loss function |
|
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 |
|