ComfyUI-NSFW-Check

ComfyUI-NSFW-Check
★ 11

图像审核NSFW检测批量评分工作流集成
检测 ComfyUI 生成图像的 NSFW 风险,返回每张图像的置信度评分,便于在工作流中自动过滤或标记不安全内容。
💡 在工作流或API中根据置信度阈值自动过滤NSFW图片。
🍴 1 Forks💻 Python🔄 2024-11-26
📦
网盘下载
复制链接后前往夸克网盘下载
https://pan.quark.cn/s/a9fb3a59e10c
📦 requirements.txt
transformers==4.37.2
pillow==10.2.0
image
📄 README

NSFW Check for ComfyUI

Project Overview

This project is designed to detect whether images generated by ComfyUI are Not Safe For Work (NSFW). It uses a machine learning model to classify images as either safe or not safe for work and returns a confidence score for the NSFW classification.

Using the score, a user can add logical filters to their workflow.

A threshold of 0.95 works well for most cases.

Usage

While one can use this node to quickly check the NSFW score of an image, it is more likely to be used when you are using workflow as an API. Using the output from this node, you can programmatically filter out NSFW images and have dynamic thresholds.

Ouput is a list of scores for the batch of images.

Credits

This project is based on ComfyUI-NSFW-Detection by trumanwong.

Install

  • Clone this repo into custom_nodes directory of ComfyUI location
  • Run pip install -r requirements.txt