Update README.md
Browse files
README.md
CHANGED
|
@@ -7,4 +7,18 @@ metrics:
|
|
| 7 |
pipeline_tag: image-classification
|
| 8 |
tags:
|
| 9 |
- art
|
| 10 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
pipeline_tag: image-classification
|
| 8 |
tags:
|
| 9 |
- art
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
The model used to predict the types of anime images, which includes the following four categories:
|
| 13 |
+
* 3D: Images rendered in 3D, including Mikumikudance, Koikatsu, etc.
|
| 14 |
+
* Bangumi: Screenshots from anime videos.
|
| 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_plus | 0.63G | 89.92% | [Confusion Matrix](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_plus/plot_confusion.png) | Model: mobilenetv3_large_100 from timm, use SCELoss as loss function |
|
| 24 |
+
| 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 |
|