deepghs/anime_classification

By deepghs · ↓ 0 · ♥ 16 · Updated 2024-10-31

What's in this repo

This page lists the files in deepghs/anime_classification, including model.onnx and more.

20 ONNX models (.onnx files)
267 files, including 182 .png, 40 .json, 20 .ckpt, 20 .onnx, 3 .0, 1 .md
Last updated 2024-10-31

ONNX files

File	Size	Download
caformer_s36/model.onnx	—	Download
caformer_s36_plus/model.onnx	—	Download
caformer_s36_v1.1_focal/model.onnx	—	Download
caformer_s36_v1.2_focal/model.onnx	—	Download
caformer_s36_v1.3_focal/model.onnx	—	Download
caformer_s36_v1.4_focal/model.onnx	—	Download
caformer_s36_v1.4_focal_fixed/model.onnx	—	Download
caformer_s36_v1.4_focal_fp32/model.onnx	—	Download
caformer_s36_v1.5_focal/model.onnx	—	Download
caformer_s36_v1/model.onnx	—	Download

Showing 10 of 20 .onnx files — see the full file list on the right.

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

The model used to predict the types of anime images, which includes the following four categories:

3D: Images rendered in 3D, including Mikumikudance, Koikatsu, etc.
Bangumi: Screenshots from anime videos.
Comic: Images of manga that contain a significant amount of text or panel sequences.
Illustration: General anime illustrations.
Not Painting: (Only available in new models) Any content that cannot be called a painting, such as artist promotional posts, game screenshots, chat logs, etc.

Name	FLOPS	Params	Accuracy	AUC	Confusion	Labels
mobilenetv3_v1.5_dist	0.63G	4.18M	94.56%	0.9946	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.5_focal	22.10G	37.22M	95.06%	0.9957	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.4_focal_fixed	22.10G	37.22M	96.21%	0.9971	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.4_focal_fp32	22.10G	37.22M	95.98%	0.9969	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
mobilenetv3_v1.4_dist	0.63G	4.18M	94.77%	0.9950	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.4_focal	22.10G	37.22M	95.82%	0.9967	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
mobilenetv3_v1.3_dist	0.63G	4.18M	96.41%	0.9973	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.3_focal	22.10G	37.22M	97.16%	0.9982	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
mobilenetv3_v1.2_dist	0.63G	4.18M	96.53%	0.9972	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.2_focal	22.10G	37.22M	97.23%	0.9982	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1.1_focal	22.10G	37.22M	95.99%	0.9967	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
mobilenetv3_v1_dist	0.63G	4.18M	94.04%	0.9928	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
caformer_s36_v1	22.10G	37.22M	94.72%	0.9934	confusion	`3d`, `bangumi`, `comic`, `illustration`, `not_painting`
mobilenetv3_dist	0.63G	4.18M	91.98%	0.9879	confusion	`3d`, `bangumi`, `comic`, `illustration`
mobilenetv3_sce_dist	0.63G	4.18M	92.35%	0.9854	confusion	`3d`, `bangumi`, `comic`, `illustration`
caformer_s36_plus	22.10G	37.22M	93.47%	0.9891	confusion	`3d`, `bangumi`, `comic`, `illustration`
mobilenetv3	0.63G	4.18M	88.96%	N/A	confusion	`3d`, `bangumi`, `comic`, `illustration`
mobilevitv2_150	9.09G	9.79M	88.21%	N/A	confusion	`3d`, `bangumi`, `comic`, `illustration`
mobilenetv3_sce	0.63G	4.18M	89.92%	0.9786	confusion	`3d`, `bangumi`, `comic`, `illustration`
caformer_s36	22.10G	37.22M	88.19%	N/A	confusion	`3d`, `bangumi`, `comic`, `illustration`

Model	FLOPs	Accuracy	Confusion Matrix	Description
caformer_s36	22.10G	88.19%	Confusion Matrix	Model: caformer_s36 from timm
caformer_s36_plus	22.10G	93.47%	Confusion Matrix	Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm
mobilenetv3	0.63G	88.96%	Confusion Matrix	Model: mobilenetv3_large_100 from timm
mobilenetv3_dist	0.63G	91.98%	Confusion Matrix	Distrillated from caformer_s36_plus, using mobilenetv3_large_100 with focal loss
mobilenetv3_sce	0.63G	89.92%	Confusion Matrix	Model: mobilenetv3_large_100 from timm, use SCELoss as loss function
mobilenetv3_sce_dist	0.63G	92.35%	Confusion Matrix	Distrillated from caformer_s36_plus, using mobilenetv3_large_100 with SCELoss
mobilevitv2_150	9.09G	88.21%	Confusion Matrix	Model: mobilevitv2_150 from timm

Model info

image-classification

tensorboard onnx art image-classification dataset:deepghs/anime_classification license:mit region:us

↓ 0 ♥ 16

Created: 2023-06-02

Updated: 2024-10-31

View on Hugging Face

Files (267)

.gitattributes

README.md

caformer_s36/meta.json

caformer_s36/metrics.json

caformer_s36/model.ckpt

caformer_s36/model.onnx ONNX

caformer_s36/plot_confusion.png

caformer_s36/plot_f1_curve.png

caformer_s36/plot_p_curve.png

caformer_s36/plot_pr_curve.png

caformer_s36/plot_r_curve.png

caformer_s36_plus/meta.json

caformer_s36_plus/metrics.json

caformer_s36_plus/model.ckpt

caformer_s36_plus/model.onnx ONNX

caformer_s36_plus/plot_confusion.png

caformer_s36_plus/plot_f1_curve.png

caformer_s36_plus/plot_p_curve.png

caformer_s36_plus/plot_pr_curve.png

caformer_s36_plus/plot_r_curve.png

caformer_s36_plus/plot_roc_curve.png

caformer_s36_v1.1_focal/meta.json

caformer_s36_v1.1_focal/metrics.json

caformer_s36_v1.1_focal/model.ckpt

caformer_s36_v1.1_focal/model.onnx ONNX

caformer_s36_v1.1_focal/plot_confusion.png

caformer_s36_v1.1_focal/plot_f1_curve.png

caformer_s36_v1.1_focal/plot_p_curve.png

caformer_s36_v1.1_focal/plot_pr_curve.png

caformer_s36_v1.1_focal/plot_r_curve.png

caformer_s36_v1.1_focal/plot_roc_curve.png

caformer_s36_v1.1_focal/plot_sample_3d.png

caformer_s36_v1.1_focal/plot_sample_bangumi.png

caformer_s36_v1.1_focal/plot_sample_comic.png

caformer_s36_v1.1_focal/plot_sample_illustration.png

caformer_s36_v1.1_focal/plot_sample_not_painting.png

caformer_s36_v1.2_focal/meta.json

caformer_s36_v1.2_focal/metrics.json

caformer_s36_v1.2_focal/model.ckpt

caformer_s36_v1.2_focal/model.onnx ONNX

caformer_s36_v1.2_focal/plot_confusion.png

caformer_s36_v1.2_focal/plot_f1_curve.png

caformer_s36_v1.2_focal/plot_p_curve.png

caformer_s36_v1.2_focal/plot_pr_curve.png

caformer_s36_v1.2_focal/plot_r_curve.png

caformer_s36_v1.2_focal/plot_roc_curve.png

caformer_s36_v1.2_focal/plot_sample_3d.png

caformer_s36_v1.2_focal/plot_sample_bangumi.png

caformer_s36_v1.2_focal/plot_sample_comic.png

caformer_s36_v1.2_focal/plot_sample_illustration.png

caformer_s36_v1.2_focal/plot_sample_not_painting.png

caformer_s36_v1.3_focal/meta.json

caformer_s36_v1.3_focal/metrics.json

caformer_s36_v1.3_focal/model.ckpt

caformer_s36_v1.3_focal/model.onnx ONNX

caformer_s36_v1.3_focal/plot_confusion.png

caformer_s36_v1.3_focal/plot_f1_curve.png

caformer_s36_v1.3_focal/plot_p_curve.png

caformer_s36_v1.3_focal/plot_pr_curve.png

caformer_s36_v1.3_focal/plot_r_curve.png

caformer_s36_v1.3_focal/plot_roc_curve.png

caformer_s36_v1.3_focal/plot_sample_3d.png

caformer_s36_v1.3_focal/plot_sample_bangumi.png

caformer_s36_v1.3_focal/plot_sample_comic.png

caformer_s36_v1.3_focal/plot_sample_illustration.png

caformer_s36_v1.3_focal/plot_sample_not_painting.png

caformer_s36_v1.4_focal/meta.json

caformer_s36_v1.4_focal/metrics.json

caformer_s36_v1.4_focal/model.ckpt

caformer_s36_v1.4_focal/model.onnx ONNX

caformer_s36_v1.4_focal/plot_confusion.png

caformer_s36_v1.4_focal/plot_f1_curve.png

caformer_s36_v1.4_focal/plot_p_curve.png

caformer_s36_v1.4_focal/plot_pr_curve.png

caformer_s36_v1.4_focal/plot_r_curve.png

caformer_s36_v1.4_focal/plot_roc_curve.png

caformer_s36_v1.4_focal/plot_sample_3d.png

caformer_s36_v1.4_focal/plot_sample_bangumi.png

caformer_s36_v1.4_focal/plot_sample_comic.png

caformer_s36_v1.4_focal/plot_sample_illustration.png

caformer_s36_v1.4_focal/plot_sample_not_painting.png

caformer_s36_v1.4_focal_fixed/events.out.tfevents.1726903602.dd381fc62bdfa9ecca29a7a9c2e134137902c3050389946c1ab5ffd4.72369.0

caformer_s36_v1.4_focal_fixed/meta.json

caformer_s36_v1.4_focal_fixed/metrics.json

caformer_s36_v1.4_focal_fixed/model.ckpt

caformer_s36_v1.4_focal_fixed/model.onnx ONNX

caformer_s36_v1.4_focal_fixed/plot_confusion.png

caformer_s36_v1.4_focal_fixed/plot_f1_curve.png

caformer_s36_v1.4_focal_fixed/plot_p_curve.png

caformer_s36_v1.4_focal_fixed/plot_pr_curve.png

caformer_s36_v1.4_focal_fixed/plot_r_curve.png

caformer_s36_v1.4_focal_fixed/plot_roc_curve.png

caformer_s36_v1.4_focal_fixed/plot_sample_3d.png

caformer_s36_v1.4_focal_fixed/plot_sample_bangumi.png

caformer_s36_v1.4_focal_fixed/plot_sample_comic.png

caformer_s36_v1.4_focal_fixed/plot_sample_illustration.png

caformer_s36_v1.4_focal_fixed/plot_sample_not_painting.png

caformer_s36_v1.4_focal_fp32/meta.json

caformer_s36_v1.4_focal_fp32/metrics.json

caformer_s36_v1.4_focal_fp32/model.ckpt

caformer_s36_v1.4_focal_fp32/model.onnx ONNX

caformer_s36_v1.4_focal_fp32/plot_confusion.png

caformer_s36_v1.4_focal_fp32/plot_f1_curve.png

caformer_s36_v1.4_focal_fp32/plot_p_curve.png

caformer_s36_v1.4_focal_fp32/plot_pr_curve.png

caformer_s36_v1.4_focal_fp32/plot_r_curve.png

caformer_s36_v1.4_focal_fp32/plot_roc_curve.png

caformer_s36_v1.4_focal_fp32/plot_sample_3d.png

caformer_s36_v1.4_focal_fp32/plot_sample_bangumi.png

caformer_s36_v1.4_focal_fp32/plot_sample_comic.png

caformer_s36_v1.4_focal_fp32/plot_sample_illustration.png

caformer_s36_v1.4_focal_fp32/plot_sample_not_painting.png

caformer_s36_v1.5_focal/events.out.tfevents.1729842384.679b526be80a17292d66892be7b862bb00288f0918687366a7497fda.39332.0

caformer_s36_v1.5_focal/meta.json

caformer_s36_v1.5_focal/metrics.json

caformer_s36_v1.5_focal/model.ckpt

caformer_s36_v1.5_focal/model.onnx ONNX

caformer_s36_v1.5_focal/plot_confusion.png

caformer_s36_v1.5_focal/plot_f1_curve.png

caformer_s36_v1.5_focal/plot_p_curve.png

caformer_s36_v1.5_focal/plot_pr_curve.png

caformer_s36_v1.5_focal/plot_r_curve.png

caformer_s36_v1.5_focal/plot_roc_curve.png

caformer_s36_v1.5_focal/plot_sample_3d.png

caformer_s36_v1.5_focal/plot_sample_bangumi.png

caformer_s36_v1.5_focal/plot_sample_comic.png

caformer_s36_v1.5_focal/plot_sample_illustration.png

caformer_s36_v1.5_focal/plot_sample_not_painting.png

caformer_s36_v1/meta.json

caformer_s36_v1/metrics.json

caformer_s36_v1/model.ckpt

caformer_s36_v1/model.onnx ONNX

caformer_s36_v1/plot_confusion.png

caformer_s36_v1/plot_f1_curve.png

caformer_s36_v1/plot_p_curve.png

caformer_s36_v1/plot_pr_curve.png

caformer_s36_v1/plot_r_curve.png

caformer_s36_v1/plot_roc_curve.png

caformer_s36_v1/plot_sample_3d.png

caformer_s36_v1/plot_sample_bangumi.png

caformer_s36_v1/plot_sample_comic.png

caformer_s36_v1/plot_sample_illustration.png

caformer_s36_v1/plot_sample_not_painting.png

mobilenetv3/meta.json

mobilenetv3/metrics.json

mobilenetv3/model.ckpt

mobilenetv3/model.onnx ONNX

mobilenetv3/plot_confusion.png

mobilenetv3/plot_f1_curve.png

mobilenetv3/plot_p_curve.png

mobilenetv3/plot_pr_curve.png

mobilenetv3/plot_r_curve.png

mobilenetv3_dist/meta.json

mobilenetv3_dist/metrics.json

mobilenetv3_dist/model.ckpt

mobilenetv3_dist/model.onnx ONNX

mobilenetv3_dist/plot_confusion.png

mobilenetv3_dist/plot_f1_curve.png

mobilenetv3_dist/plot_p_curve.png

mobilenetv3_dist/plot_pr_curve.png

mobilenetv3_dist/plot_r_curve.png

mobilenetv3_dist/plot_roc_curve.png

mobilenetv3_sce/meta.json

mobilenetv3_sce/metrics.json

mobilenetv3_sce/model.ckpt

mobilenetv3_sce/model.onnx ONNX

mobilenetv3_sce/plot_confusion.png

mobilenetv3_sce/plot_f1_curve.png

mobilenetv3_sce/plot_p_curve.png

mobilenetv3_sce/plot_pr_curve.png

mobilenetv3_sce/plot_r_curve.png

mobilenetv3_sce/plot_roc_curve.png

mobilenetv3_sce_dist/meta.json

mobilenetv3_sce_dist/metrics.json

mobilenetv3_sce_dist/model.ckpt

mobilenetv3_sce_dist/model.onnx ONNX

mobilenetv3_sce_dist/plot_confusion.png

mobilenetv3_sce_dist/plot_f1_curve.png

mobilenetv3_sce_dist/plot_p_curve.png

mobilenetv3_sce_dist/plot_pr_curve.png

mobilenetv3_sce_dist/plot_r_curve.png

mobilenetv3_sce_dist/plot_roc_curve.png

mobilenetv3_v1.2_dist/meta.json

mobilenetv3_v1.2_dist/metrics.json

mobilenetv3_v1.2_dist/model.ckpt

mobilenetv3_v1.2_dist/model.onnx ONNX

mobilenetv3_v1.2_dist/plot_confusion.png

mobilenetv3_v1.2_dist/plot_f1_curve.png

mobilenetv3_v1.2_dist/plot_p_curve.png

mobilenetv3_v1.2_dist/plot_pr_curve.png

mobilenetv3_v1.2_dist/plot_r_curve.png

mobilenetv3_v1.2_dist/plot_roc_curve.png

mobilenetv3_v1.2_dist/plot_sample_3d.png

mobilenetv3_v1.2_dist/plot_sample_bangumi.png

mobilenetv3_v1.2_dist/plot_sample_comic.png

mobilenetv3_v1.2_dist/plot_sample_illustration.png

mobilenetv3_v1.2_dist/plot_sample_not_painting.png

mobilenetv3_v1.3_dist/meta.json

mobilenetv3_v1.3_dist/metrics.json

mobilenetv3_v1.3_dist/model.ckpt

mobilenetv3_v1.3_dist/model.onnx ONNX

mobilenetv3_v1.3_dist/plot_confusion.png

mobilenetv3_v1.3_dist/plot_f1_curve.png

mobilenetv3_v1.3_dist/plot_p_curve.png

mobilenetv3_v1.3_dist/plot_pr_curve.png

mobilenetv3_v1.3_dist/plot_r_curve.png

mobilenetv3_v1.3_dist/plot_roc_curve.png

mobilenetv3_v1.3_dist/plot_sample_3d.png

mobilenetv3_v1.3_dist/plot_sample_bangumi.png

mobilenetv3_v1.3_dist/plot_sample_comic.png

mobilenetv3_v1.3_dist/plot_sample_illustration.png

mobilenetv3_v1.3_dist/plot_sample_not_painting.png

mobilenetv3_v1.4_dist/meta.json

mobilenetv3_v1.4_dist/metrics.json

mobilenetv3_v1.4_dist/model.ckpt

mobilenetv3_v1.4_dist/model.onnx ONNX

mobilenetv3_v1.4_dist/plot_confusion.png

mobilenetv3_v1.4_dist/plot_f1_curve.png

mobilenetv3_v1.4_dist/plot_p_curve.png

mobilenetv3_v1.4_dist/plot_pr_curve.png

mobilenetv3_v1.4_dist/plot_r_curve.png

mobilenetv3_v1.4_dist/plot_roc_curve.png

mobilenetv3_v1.4_dist/plot_sample_3d.png

mobilenetv3_v1.4_dist/plot_sample_bangumi.png

mobilenetv3_v1.4_dist/plot_sample_comic.png

mobilenetv3_v1.4_dist/plot_sample_illustration.png

mobilenetv3_v1.4_dist/plot_sample_not_painting.png

mobilenetv3_v1.5_dist/events.out.tfevents.1730142360.679b526be80a17292d66892be7b862bb00288f0918687366a7497fda.2270384.0

mobilenetv3_v1.5_dist/meta.json

mobilenetv3_v1.5_dist/metrics.json

mobilenetv3_v1.5_dist/model.ckpt

mobilenetv3_v1.5_dist/model.onnx ONNX

mobilenetv3_v1.5_dist/plot_confusion.png

mobilenetv3_v1.5_dist/plot_f1_curve.png

mobilenetv3_v1.5_dist/plot_p_curve.png

mobilenetv3_v1.5_dist/plot_pr_curve.png

mobilenetv3_v1.5_dist/plot_r_curve.png

mobilenetv3_v1.5_dist/plot_roc_curve.png

mobilenetv3_v1.5_dist/plot_sample_3d.png

mobilenetv3_v1.5_dist/plot_sample_bangumi.png

mobilenetv3_v1.5_dist/plot_sample_comic.png

mobilenetv3_v1.5_dist/plot_sample_illustration.png

mobilenetv3_v1.5_dist/plot_sample_not_painting.png

mobilenetv3_v1_dist/meta.json

mobilenetv3_v1_dist/metrics.json

mobilenetv3_v1_dist/model.ckpt

mobilenetv3_v1_dist/model.onnx ONNX

mobilenetv3_v1_dist/plot_confusion.png

mobilenetv3_v1_dist/plot_f1_curve.png

mobilenetv3_v1_dist/plot_p_curve.png

mobilenetv3_v1_dist/plot_pr_curve.png

mobilenetv3_v1_dist/plot_r_curve.png

mobilenetv3_v1_dist/plot_roc_curve.png

mobilenetv3_v1_dist/plot_sample_3d.png

mobilenetv3_v1_dist/plot_sample_bangumi.png

mobilenetv3_v1_dist/plot_sample_comic.png

mobilenetv3_v1_dist/plot_sample_illustration.png

mobilenetv3_v1_dist/plot_sample_not_painting.png

mobilevitv2_150/meta.json

mobilevitv2_150/metrics.json

mobilevitv2_150/model.ckpt

mobilevitv2_150/model.onnx ONNX

mobilevitv2_150/plot_confusion.png

mobilevitv2_150/plot_f1_curve.png

mobilevitv2_150/plot_p_curve.png

mobilevitv2_150/plot_pr_curve.png

mobilevitv2_150/plot_r_curve.png

Related models

More image-classification models

Browse all image-classification models