Skip to content

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

Notifications You must be signed in to change notification settings

phyblas/nsfw-shorier_comfyui

Repository files navigation

nsfw-shorier_comfyui

(→ English Description) (→ 日本語解説はQiitaで) (→ อ่านภาษาไทย

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

安装

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

支持的模型

可以选择用于识别 NSFW 内容的检测模型:

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

首次使用时,系统会自动下载模型文件,请耐心等待下载完成。后续使用时会直接加载已下载的模型,无需重复下载。

节点

此扩展包提供 9 个功能节点。

- GetNsfwScore

基础检测节点,输出 NSFW 内容的可能性分数。分数范围为 0 到 1,分数越高,包含 NSFW 内容的可能性越大。

不同模型的判定标准可能存在差异,输出的分数也会有所不同。

- IsNsfw

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

- ReplaceIfNsfw

使用自己准备的图片替换被识别为 NSFW 的图片。若将 resize 设置为 true,替换图片会调整尺寸以匹配原图。

- FilterNsfw

对 NSFW 图片进行处理,支持多种处理模式,包括模糊化、马赛克处理,或替换为全黑、全白等效果。

- FilterNsfwWithText

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

- SaveImageSfw

功能类似于原生 SaveImage 节点,用于保存图片,但会自动将 NSFW 内容转换为黑色。支持保存为 jpgwebp 格式。此节点改编自 comfyui-saveimage-plus

- FindNudenetPart, FindNsfwPart

使用 nudenet 模型识别图片中的特定元素。FindNudenetPart 会检测所有元素,FindNsfwPart 则专门识别通常被视为 NSFW 的内容。也支持阈值设定。

此节点改编自 ComfyUI-utils-nodes

nudenet 模型提供 320px 和 640px 两种版本。640px 版本能检测更细小的元素,但模型文件更大,计算时间更长。

这两种 nudenet 模型也可用于 GetNsfwScore 及其他节点,但其输出的分数仅代表检测到的最高分元素,且不提供元素位置信息。

- CensorNsfwPart

FindNsfwPart 类似,先识别出 NSFW 部分,然后对这些区域进行特定处理。支持的处理方式与 FilterNsfw 相同,但仅应用于识别出的特定区域。

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published