ComfyUI Extension: ComfyUI-NSFW-Detection

Authored by trumanwong

Created

Updated

50 stars

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

An implementation of NSFW Detection for ComfyUI

Looking for a different extension?

Custom Nodes (1)

README

This is an implementation of NSFW Detection 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. If an image is classified as NSFW, an alternative image is returned.

Install

  1. Clone this repo into custom_nodes directory of ComfyUI location

  2. Run pip install -r requirements.txt

Usage

The main functionality of the project is encapsulated in the NSFWDetection class in the node.py file. This class has a run method that takes three parameters:

  • image: The image to be classified.
  • score: The threshold score for classifying an image as NSFW.
  • alternative_image: The image to be returned if the input image is classified as NSFW.

Example

<img src="images/example.png" raw=true>

https://github.com/trumanwong/ComfyUI-NSFW-Detection/blob/main/workflow.json

Contributing

Contributions are welcome. Please submit a pull request if you have any improvements or bug fixes.

License

This project is licensed under the terms of the MIT license.

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

Learn more