anti-nsfw 撤回 nsfw 图片

koishi-plugin-anti-nsfw

npm

撤回 nsfw 图片

介绍

该插件会在你的 Koishi 中运行一个 onnx

预测收到的图片是否为 nsfw 图片,并撤回 nsfw 图片

预测的准确率为 0.9654

配置

下载模型

如果您的机器可以访问 huggingface, 则无需做此步骤

点我查看👈

前往 huggingface 下载模型

同时,我将模型上传到了魔塔社区,如果访问不了 huggingface 可以从这里:point_left:下载

下载模型配置文件

找一个存放模型的目录 例如 D:\models\AdamCodd\vit-base-nsfw-detector

你需要下载这些文件,保存到 vit-base-nsfw-detector 文件夹

  • config.json

  • preprocessor_config.json

下载 onnx

新建一个文件夹 命名为 onnx, 将以下模型放入 onnx 文件夹

  • model_quantized.onnx

配置插件

如果您的机器可以访问 huggingface, 则无需做此步骤

点我查看👈

填写存放模型的路径 例如 D:\models

填写模型名称 AdamCodd\vit-base-nsfw-detector

启用 transformers 插件和 anti-nsfw 插件

2 个赞