File size: 782 Bytes
3f7ba57 9c6ceae 3f7ba57 594ddb4 9c6ceae 594ddb4 9c6ceae 9dc62e7 9c6ceae 594ddb4 24eeb0f 9c6ceae 38f5bdc 9c6ceae 24eeb0f |
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 |
---
license: mit
pipeline_tag: image-classification
---
A classification using mmpretrain trained to classify western images based on ConvNeXtV2-tiny.Used for classifying anime images based on whether they are in the Western style.
The evaluation accuracy on the validation set is 95%.
Trained using 7,000 Western images and 8,000 non-Western images, with the Western training set sampled from e-hentai.
Of course, this model also has many shortcomings, such as a very low recognition accuracy for line-drawing images.
Huggingface space:https://huggingface.co/spaces/TLME/western-anime-images-classification
# How to use
Python>=3.9
```
Install pytorch
pip install -r requirements.txt
edit infer.py , change "path = './testimg/'" to your target folder
python infer.py
``` |