File size: 2,360 Bytes
94ff89f 1d47278 32d6c8b 6358d59 32d6c8b 6358d59 fbcf2a3 32d6c8b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
---
license: openrail
metrics:
- accuracy
pipeline_tag: image-classification
tags:
- art
datasets:
- deepghs/ai_image_corrupted
library_name: dghs-imgutils
---
This is the classifier model for predicting the anime-style stable-diffusion-generated images are corrupted or not.
Trained on dataset [deepghs/ai_image_corrupted](https://huggingface.co/datasets/deepghs/ai_image_corrupted).
## Models
| Name | FLOPS | Params | Accuracy | AUC | Confusion | Labels |
|:-------------------------:|:-------:|:--------:|:----------:|:------:|:---------------------------------------------------------------------------------------------------------------------:|:---------------------:|
| caformer_s36_v0_focal | 22.10G | 37.21M | 95.65% | 0.9916 | [confusion](https://huggingface.co/deepghs/ai_image_corrupted/blob/main/caformer_s36_v0_focal/plot_confusion.png) | `corrupted`, `normal` |
| caformer_s36_v0_sce | 22.10G | 37.21M | 95.77% | 0.9894 | [confusion](https://huggingface.co/deepghs/ai_image_corrupted/blob/main/caformer_s36_v0_sce/plot_confusion.png) | `corrupted`, `normal` |
| mobilenetv3_v0_focal_dist | 0.63G | 4.18M | 94.02% | 0.9842 | [confusion](https://huggingface.co/deepghs/ai_image_corrupted/blob/main/mobilenetv3_v0_focal_dist/plot_confusion.png) | `corrupted`, `normal` |
## How to Use
You can use this model with [dghs-imgutils](https://dghs-imgutils.deepghs.org/main/api_doc/generic/classify.html).
```python
from imgutils.generic import classify_predict_score
classify_predict_score(
'sample_image.png',
repo_id='deepghs/ai_image_corrupted',
model_name='mobilenetv3_v0_focal_dist',
)
# {'corrupted': 0.7807788848876953, 'normal': 0.2192210853099823}
```
## Note
This model is trained on SD1.5 images, generated by 8 different base models.
So the better solution for this problem is to use some metrics like artstyle embeddings,
this model will be deprecated as soon as the artstyle embedding model for anime images is completed.
## Citation
```
@misc{Citation,
title={AI-Corrupt Score for Anime Images},
author={narugo1992},
year={2023},
howpublished={\url{https://huggingface.co/deepghs/ai_image_corrupted}}
}
```
|