许可证：Apache-2.0
任务类型：图像分类

关于

此项目是MichalMlodawski/nsfw-image-detection-large的一个分支，原项目已不可用。

使用示例

from PIL import Image
import torch
from transformers import AutoProcessor, FocalNetForImageClassification

DEVICE = torch.device("cuda")
model_path = "lovetillion/nsfw-image-detection-large"

# 加载模型和特征提取器
feature_extractor = AutoProcessor.from_pretrained(model_path)
model = FocalNetForImageClassification.from_pretrained(model_path).to(DEVICE)
model.eval()

# 模型标签到NSFW类别的映射
label_to_category = {
    "LABEL_0": "安全",
    "LABEL_1": "可疑",
    "LABEL_2": "不安全"
}

filename = "example.png"
image = Image.open(filename)
inputs = feature_extractor(images=image, return_tensors="pt")
inputs.to(DEVICE)

with torch.no_grad():
    outputs = model(**inputs)
    probabilities = torch.nn.functional.softmax(outputs.logits, dim=-1)
    confidence, predicted = torch.max(probabilities, 1)
label = model.config.id2label[predicted.item()]

if label != "SAFE":
    print( label, confidence.item() * 100, filename )
else:
    print( label, confidence.item() * 100, filename )

更多信息

• 生产级集成工作流实时演示：https://piglet.video
• 伦理AI白皮书研究成果：https://lovetillion.org/liaise.pdf
• 加入我们的Telegram频道：https://t.me/pigletproject