narugo commited on
Commit
3abc604
·
verified ·
1 Parent(s): 7c7f42a

Sync README for deepghs/anime_classification, on 2024-09-23 16:35:46 UTC

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -18,7 +18,7 @@ The model used to predict the types of anime images, which includes the followin
18
 
19
  | Name | FLOPS | Params | Accuracy | AUC | Confusion | Labels |
20
  |:-----------------------------:|:-------:|:--------:|:----------:|:------:|:---------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------:|
21
- | caformer_s36_v1.4_focal_fixed | 22.10G | 37.22M | 95.85% | 0.9968 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal_fixed/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
22
  | caformer_s36_v1.4_focal_fp32 | 22.10G | 37.22M | 95.98% | 0.9969 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal_fp32/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
23
  | mobilenetv3_v1.4_dist | 0.63G | 4.18M | 94.77% | 0.9950 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_v1.4_dist/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
24
  | caformer_s36_v1.4_focal | 22.10G | 37.22M | 95.82% | 0.9967 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
@@ -45,4 +45,5 @@ The model used to predict the types of anime images, which includes the followin
45
  | 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 |
46
  | 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 |
47
  | 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 |
48
- | 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 |
 
 
18
 
19
  | Name | FLOPS | Params | Accuracy | AUC | Confusion | Labels |
20
  |:-----------------------------:|:-------:|:--------:|:----------:|:------:|:---------------------------------------------------------------------------------------------------------------------------:|:--------------------------------------------------------:|
21
+ | caformer_s36_v1.4_focal_fixed | 22.10G | 37.22M | 96.08% | 0.9970 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal_fixed/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
22
  | caformer_s36_v1.4_focal_fp32 | 22.10G | 37.22M | 95.98% | 0.9969 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal_fp32/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
23
  | mobilenetv3_v1.4_dist | 0.63G | 4.18M | 94.77% | 0.9950 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/mobilenetv3_v1.4_dist/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
24
  | caformer_s36_v1.4_focal | 22.10G | 37.22M | 95.82% | 0.9967 | [confusion](https://huggingface.co/deepghs/anime_classification/blob/main/caformer_s36_v1.4_focal/plot_confusion.png) | `3d`, `bangumi`, `comic`, `illustration`, `not_painting` |
 
45
  | 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 |
46
  | 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 |
47
  | 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 |
48
+ | 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 |
49
+