deepghs/anime_rating
By deepghs · ↓ 0 · ♥ 3 · Updated 2024-01-19
What's in this repo
This page lists the files in deepghs/anime_rating, including model.onnx and more.
- 7 ONNX models (.onnx files)
- 80 files, including 50 .png, 14 .json, 7 .ckpt, 7 .onnx, 1 .md
- Last updated 2024-01-19
ONNX files
| File | Size | Download |
|---|---|---|
| caformer_b36_v1_pruned_ls0.1/model.onnx | — | Download |
| caformer_s36_plus/model.onnx | — | Download |
| caformer_s36_v1_pruned_ls0.1/model.onnx | — | Download |
| mobilenetv3/model.onnx | — | Download |
| mobilenetv3_sce/model.onnx | — | Download |
| mobilenetv3_sce_dist/model.onnx | — | Download |
| mobilenetv3_v1_pruned_ls0.1/model.onnx | — | Download |
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
Image rating (safe, r15, r18) prediction model.
Please note that due to the often unclear boundaries between safe, r15 and r18 levels, there is no objective ground truth for this task, and the data is scraped without any manual filtering. Therefore, the models trained on this dataset can only provide rough judgments. If you require an accurate solution for classifying R18 images, it is recommended to consider a solution based on keypoint object detection.
| Model | FLOPs | Accuracy | Confusion Matrix | Description |
|---|---|---|---|---|
| caformer_s36_plus | 22.10G | 74.26% | Confusion Matrix | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm |
| mobilenetv3 | 0.63G | 64.77% | Confusion Matrix | Model: mobilenetv3_large_100 from timm |
| mobilenetv3_sce | 0.63G | 66.27% | Confusion Matrix | Model: mobilenetv3_large_100 from timm, use SCELoss as loss function |
| mobilenetv3_sce_dist | 0.63G | 69.49% | Confusion Matrix | Distillated from caformer_s36_plus, using mobilenetv3_large_100 |
Model info
onnx
art
image-classification
dataset:deepghs/anime_rating
license:mit
region:us
↓ 0
♥ 3
Created: 2023-06-03
Updated: 2024-01-19
View on Hugging FaceFiles (80)
.gitattributes
README.md
caformer_b36_v1_pruned_ls0.1/meta.json
caformer_b36_v1_pruned_ls0.1/metrics.json
caformer_b36_v1_pruned_ls0.1/model.ckpt
caformer_b36_v1_pruned_ls0.1/model.onnx
ONNX
caformer_b36_v1_pruned_ls0.1/plot_confusion.png
caformer_b36_v1_pruned_ls0.1/plot_f1_curve.png
caformer_b36_v1_pruned_ls0.1/plot_p_curve.png
caformer_b36_v1_pruned_ls0.1/plot_pr_curve.png
caformer_b36_v1_pruned_ls0.1/plot_r_curve.png
caformer_b36_v1_pruned_ls0.1/plot_roc_curve.png
caformer_b36_v1_pruned_ls0.1/plot_sample_r15.png
caformer_b36_v1_pruned_ls0.1/plot_sample_r18.png
caformer_b36_v1_pruned_ls0.1/plot_sample_safe.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_pruned_ls0.1/meta.json
caformer_s36_v1_pruned_ls0.1/metrics.json
caformer_s36_v1_pruned_ls0.1/model.ckpt
caformer_s36_v1_pruned_ls0.1/model.onnx
ONNX
caformer_s36_v1_pruned_ls0.1/plot_confusion.png
caformer_s36_v1_pruned_ls0.1/plot_f1_curve.png
caformer_s36_v1_pruned_ls0.1/plot_p_curve.png
caformer_s36_v1_pruned_ls0.1/plot_pr_curve.png
caformer_s36_v1_pruned_ls0.1/plot_r_curve.png
caformer_s36_v1_pruned_ls0.1/plot_roc_curve.png
caformer_s36_v1_pruned_ls0.1/plot_sample_r15.png
caformer_s36_v1_pruned_ls0.1/plot_sample_r18.png
caformer_s36_v1_pruned_ls0.1/plot_sample_safe.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_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_pruned_ls0.1/meta.json
mobilenetv3_v1_pruned_ls0.1/metrics.json
mobilenetv3_v1_pruned_ls0.1/model.ckpt
mobilenetv3_v1_pruned_ls0.1/model.onnx
ONNX
mobilenetv3_v1_pruned_ls0.1/plot_confusion.png
mobilenetv3_v1_pruned_ls0.1/plot_f1_curve.png
mobilenetv3_v1_pruned_ls0.1/plot_p_curve.png
mobilenetv3_v1_pruned_ls0.1/plot_pr_curve.png
mobilenetv3_v1_pruned_ls0.1/plot_r_curve.png
mobilenetv3_v1_pruned_ls0.1/plot_roc_curve.png
mobilenetv3_v1_pruned_ls0.1/plot_sample_r15.png
mobilenetv3_v1_pruned_ls0.1/plot_sample_r18.png
mobilenetv3_v1_pruned_ls0.1/plot_sample_safe.png