|
--- |
|
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}} |
|
} |
|
``` |
|
|
