Update README.md
Browse files
README.md
CHANGED
|
@@ -14,16 +14,13 @@ license: mit
|
|
| 14 |
Images are first divided into patches and classified as rich texture or poor texture. The azimuthal integral is then calculated for each patch with positional embedding applied. See train.py for subsequent classifier architecture.
|
| 15 |
The four datasets I trained on:
|
| 16 |
|
| 17 |
-
|
| 18 |
- AI Recognition Dataset: https://www.kaggle.com/datasets/superpotato9/dalle-recognition-dataset
|
| 19 |
- AI ArtBench: https://www.kaggle.com/datasets/ravidussilva/real-ai-art
|
| 20 |
- Dalle-3 Images: https://huggingface.co/datasets/ehristoforu/dalle-3-images
|
| 21 |
- 130k Images--Universal Image Embeddings: https://www.kaggle.com/datasets/rhtsingh/130k-images-512x512-universal-image-embeddings
|
| 22 |
|
| 23 |
-
|
| 24 |
I sampled approximated 50,000 real and 50,000 AI generated images. I was able to achieve 97.5% validation accuracy
|
| 25 |
|
| 26 |
-
|
| 27 |
# Thanks
|
| 28 |
The model design is inspired by https://arxiv.org/abs/2311.12397 and https://arxiv.org/abs/2003.01826.
|
| 29 |
Used code from https://github.com/hridayK/Detection-of-AI-generated-images and https://github.com/cc-hpc-itwm/UpConv
|
|
|
|
| 14 |
Images are first divided into patches and classified as rich texture or poor texture. The azimuthal integral is then calculated for each patch with positional embedding applied. See train.py for subsequent classifier architecture.
|
| 15 |
The four datasets I trained on:
|
| 16 |
|
|
|
|
| 17 |
- AI Recognition Dataset: https://www.kaggle.com/datasets/superpotato9/dalle-recognition-dataset
|
| 18 |
- AI ArtBench: https://www.kaggle.com/datasets/ravidussilva/real-ai-art
|
| 19 |
- Dalle-3 Images: https://huggingface.co/datasets/ehristoforu/dalle-3-images
|
| 20 |
- 130k Images--Universal Image Embeddings: https://www.kaggle.com/datasets/rhtsingh/130k-images-512x512-universal-image-embeddings
|
| 21 |
|
|
|
|
| 22 |
I sampled approximated 50,000 real and 50,000 AI generated images. I was able to achieve 97.5% validation accuracy
|
| 23 |
|
|
|
|
| 24 |
# Thanks
|
| 25 |
The model design is inspired by https://arxiv.org/abs/2311.12397 and https://arxiv.org/abs/2003.01826.
|
| 26 |
Used code from https://github.com/hridayK/Detection-of-AI-generated-images and https://github.com/cc-hpc-itwm/UpConv
|