Aiornot

🚀 圖像分類模型

本模型用於對真實圖像和AI生成圖像進行分類。它基於在aiornot數據集上微調的swin-tiny-patch4-window7-224模型構建。

🚀 快速開始

本模型可用於區分真實圖像和AI生成圖像。它基於在aiornot數據集上微調的swin-tiny-patch4-window7-224模型。

以下是使用該模型的示例代碼：

import torch
from transformers import AutoFeatureExtractor, AutoModelForImageClassification

labels = ["Real", "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] 

📄 許可證

本項目採用Apache-2.0許可證。

📦 模型信息