ComfyUI Extension: nsfw-shorier_comfyui

Authored by phyblas

Created

Updated

3 stars

Performs various processing on images containing NSFW content within ComfyUI. The model used for detecting NSFW content can be selected.

Custom Nodes (0)

    README

    nsfw-shorier_comfyui

    (→ english description) (→ 日本語の説明

    这个包用于在 ComfyUI 中处理包含 NSFW(Not Safe For Work,不适宜在工作场所观看)内容的图片。其名称来源于日语中的“処理”(shori,意为“处理”)。

    安装

    将此仓库放置在 ComfyUI 的 ComfyUI/custom_nodes/ 文件夹中。也可以通过 ComfyUI Manager 进行安装。

    使用的模型

    可以选择用于判断 NSFW 内容的模型:

    compvis 是默认模型,根据我的经验,它能提供最准确的结果。但也可以尝试其他模型,请选择自己认为效果最好的一个。

    在首次使用时,系统会自动下载模型文件,请耐心等待片刻。之后再次使用即可快速加载,无需重复下载。

    节点

    这个包提供了 6 个节点。

    - GetNsfwScore

    这是最基本的节点,仅用于输出表示 NSFW 可能性的分数。分数范围在 0 到 1 之间,分数越高,越有可能是 NSFW 内容。

    不同模型的判断标准可能不同,有时会给出差异较大的分数。

    - IsNsfw

    通过指定 threshold(阈值)来判断是否为 NSFW 内容。threshold 越低,越容易被判断为 NSFW。

    - ReplaceIfNsfw

    使用自己准备的图片替换被识别为 NSFW 的图片。如果 resize 设置为true,则会调整替换图片的大小以匹配输入图片的尺寸。

    - FilterNsfw

    对 NSFW 图片进行处理。可以选择处理模式,例如模糊化、马赛克处理,或直接替换为全黑、全白等。

    - FilterNsfwWithText

    功能与 FilterNsfw 相同,但额外支持添加文本。该方法基于 ComfyUI-TextOverlay 包。可以使用自己电脑中的字体,也可以使用自定义字体文件,例如 字魂手刻宋.ttfmeiryoub.ttc。将字体文件放在本仓库的 font 文件夹中即可使用。如果已经安装了 ComfyUI-TextOverlay 包,也可以使用该包 fonts 文件夹中的字体文件。

    - SaveImageSfw

    与原始的 SaveImage 节点类似,用于保存图片,但如果图片被识别为 NSFW,则会将其变为黑色。还可以选择将图片保存为 jpgwebp 格式。该节点改编自 comfyui-saveimage-plus