许可证：apache-2.0
数据集：

• competitions/aiornot
  语言：
• 英文
  评估指标：
• 准确率
  标签：
• 生成式AI
• 分类

用于区分真实图像与AI生成图像的分类模型。
该模型基于swin-tiny-patch4-window7-224架构，并在aiornot数据集上进行了微调。
使用方法：

import torch
from transformers import AutoFeatureExtractor, AutoModelForImageClassification

labels = ["真实", "AI"]
feature_extractor = AutoFeatureExtractor.from_pretrained("Nahrawy/AIorNot")
model = AutoModelForImageClassification.from_pretrained("Nahrawy/AIorNot")

input = feature_extractor(image, return_tensors="pt")
with torch.no_grad():
  outputs = model(**input)
  logits = outputs.logits
prediction = logits.argmax(-1).item()
label = labels[prediction]