Aiornot

🚀 画像分類モデル (AI生成画像と実画像)

このモデルは、実画像とAI生成画像を分類するための分類モデルです。Swin-Tinyモデルをaiornotデータセットでファインチューニングしています。

🚀 クイックスタート

このモデルを使用するには、以下のコードを実行します。

基本的な使用法

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ライセンスの下で公開されています。