NYUAD AI Generated Images Detector

🚀 AI-generated_images_detector

このモデルは、AI生成画像の検出に特化した画像分類モデルです。評価セットにおいて高い精度を達成しており、画像のソースがAI生成かどうかを効果的に判断できます。

🚀 クイックスタート

このモデルを使用するには、以下の手順に従ってください。

from PIL import Image
from transformers import pipeline
classifier = pipeline("image-classification", model="NYUAD-ComNets/NYUAD_AI-generated_images_detector")
image=Image.open("path_to_image")
pred=classifier(image)
print(pred)

✨ 主な機能

• 画像分類タスクに特化し、AI生成画像を高精度で検出します。
• 評価セットにおいて、損失率0.0987、精度0.9736を達成しています。

📦 インストール

このモデルを使用するには、transformersとPILライブラリが必要です。以下のコマンドでインストールできます。

pip install transformers pillow

💻 使用例

基本的な使用法

from PIL import Image
from transformers import pipeline
classifier = pipeline("image-classification", model="NYUAD-ComNets/NYUAD_AI-generated_images_detector")
image=Image.open("path_to_image")
pred=classifier(image)
print(pred)

📚 ドキュメント

評価結果

このモデルは、評価セットで以下の結果を達成しています。

• 損失率: 0.0987
• 精度: 0.9736

学習結果

📄 ライセンス

このモデルは、Apache 2.0ライセンスの下で提供されています。

📚 引用情報

@article{aldahoul2024detecting,
  title={Detecting AI-Generated Images Using Vision Transformers: A Robust Approach for Safeguarding Visual Media Integrity},
  author={AlDahoul, Nouar and Zaki, Yasir},
  journal={Available at SSRN},
  year={2024}
}

@misc{ComNets,
      url={https://huggingface.co/NYUAD-ComNets/NYUAD_AI-generated_images_detector](https://huggingface.co/NYUAD-ComNets/NYUAD_AI-generated_images_detector)},
      title={NYUAD_AI-generated_images_detector},
      author={Nouar AlDahoul, Yasir Zaki}
}