ComfyUI NSFW Detection

這是對Comfyui的NSFW檢測的實現

項目概述

該項目旨在檢測comfyui生成的圖像是否不安全（NSFW）。它使用機器學習模型將圖像分類為安全或不安全的工作。如果將圖像歸類為NSFW，則返回替代圖像。

安裝

1. 克隆此倉庫到custo_nodes comfyui位置的目錄

2. 運行PIP安裝-R要求

用法

該項目的主要功能封裝在node.py文件中的NSFWDetection類中。該類有一個run方法，該方法採用三個參數：

• image ：要分類的圖像。
• score ：將圖像分類為NSFW的閾值分數。
• alternative_image ：如果輸入圖像分類為NSFW，則要返回的圖像。

例子

https://github.com/trumanwong/comfyui-nsfw-detection/blob/main/workflow.json

貢獻

歡迎捐款。如果您有任何改進或錯誤修復，請提交拉動請求。

執照

該項目是根據MIT許可證的條款獲得許可的。