AI ImageClassification MidjourneyV6 SDXL

🚀 AI图像分类 - Midjourney V6 & SDXL 模型卡片

本模型旨在区分AI生成图像与人类创作图像，特别是针对Midjourney V6和Stable Diffusion XL (SDXL)的输出图像，为内容审核、事实核查等场景提供了有效的解决方案。

🚀 快速开始

你可以使用🤗 Transformers库来使用此模型：

from transformers import AutoModelForImageClassification, AutoFeatureExtractor
from PIL import Image
import torch

# 加载模型和特征提取器
model_name = "ideepankarsharma2003/AI_ImageClassification_MidjourneyV6_SDXL"
model = AutoModelForImageClassification.from_pretrained(model_name)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)

# 加载并预处理图像
image = Image.open("path_to_image.jpg")
inputs = feature_extractor(images=image, return_tensors="pt")

# 进行推理
with torch.no_grad():
    outputs = model(**inputs)
    logits = outputs.logits
    predicted_label = logits.argmax(-1).item()

# 标签映射
id2label = {0: "ai_gen", 1: "human"}
print("预测标签:", id2label[predicted_label])

✨ 主要特性

• 基于Swin Transformer架构，能够有效区分AI生成和人类创作的图像。
• 专门针对Midjourney V6和Stable Diffusion XL (SDXL)的输出图像进行训练。
• 可用于内容审核、事实核查和合成媒体检测等场景。

📦 安装指南

文档未提供安装步骤，可参考🤗 Transformers库的官方文档进行安装。

📚 详细文档

模型详情

模型描述

此模型是一个基于Swin Transformer的分类器，旨在区分AI生成和人类创作的图像，特别关注Midjourney V6和**Stable Diffusion XL (SDXL)**的输出。它在经过精心策划的AI生成图像数据集上进行了训练。

模型来源

• 仓库地址：Hugging Face模型仓库

用途

直接用途

此模型可用于检测Midjourney V6和SDXL生成的AI图像。它适用于内容审核、事实核查和合成媒体检测。

非适用场景

• 该模型并非设计用于检测所有生成模型生成的AI图像。
• 对于经过大量编辑的AI生成图像或混合了人类元素的图像，其性能可能不佳。
• 它不用于法医级别的深度伪造检测。

偏差、风险和局限性

此模型专门在Midjourney V6和Stable Diffusion XL数据集上进行训练。它可能无法很好地泛化到其他AI模型生成的图像。此外，数据集中的偏差可能导致误报（将真实图像标记为AI生成）或漏报（未能检测到AI生成的内容）。

建议

用户应使用其他工具验证结果，不要仅依赖此模型进行高风险决策。在部署之前，应在特定领域的数据集上测试模型性能。

训练详情

训练数据

该模型在以下数据集上进行训练：

• ImageClassificationStableDiffusion_small
• Midjourney_v6_Classification_small_shuffled
• AIGeneratedImages_Midjourney

训练过程

• 图像大小：224x224
• 补丁大小：4
• 嵌入维度：128
• 层数：4
• 每阶段注意力头数：[4, 8, 16, 32]
• 丢弃率：
  • 注意力：0.0
  • 隐藏层：0.0
  • 随机丢弃路径：0.1
• 激活函数：GeLU
• 优化器：AdamW
• 学习率调度器：Cosine Annealing
• 精度：float32
• 训练步数：3414

评估

测试数据、因素和指标

测试数据

该模型在训练数据集的单独验证分割上进行评估。

指标

• 准确率
• 精确率和召回率
• F1分数

总结

该模型能够有效区分AI生成和人类创作的图像，但其性能可能会受到数据集偏差和分布外示例的影响。

引用

如果使用此模型，请引用：

@misc{ai_image_classification,
  author = {Deepankar Sharma},
  title = {AI Image Classification - Midjourney V6 & SDXL},
  year = {2024},
  publisher = {Hugging Face},
  howpublished = {\url{https://huggingface.co/ideepankarsharma2003/AI_ImageClassification_MidjourneyV6_SDXL}}
}

模型卡片作者

• 作者：Deepankar Sharma