Nahrawy/AIorNot

Nahrawy

Clasificación de imagen

Modelo de clasificación utilizado para clasificar imágenes reales e imágenes generadas por IA. El modelo utilizado es swin-tiny-patch4-window7-224 afinado en el conjunto de datos aiornot.

Como usar

Para usar el modelo:
import torch
from transformers import AutoFeatureExtractor, AutoModelForImageClassification

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

Funcionalidades

Clasificación de imágenes
Transformers
PyTorch
Redes neuronales
AI generativa
Clasificación

Casos de uso

Clasificación de imágenes reales vs. generadas por IA
Filtración de contenido generado por IA