notAI-tech / NudeNet

Lightweight nudity detection
https://nudenet.notai.tech/
GNU Affero General Public License v3.0
1.66k stars 335 forks source link

I made a censor extension for stable diffusion web ui using nudenet #131

Closed w-e-w closed 7 months ago

w-e-w commented 8 months ago

just want to let you know that I made a censor extension for stable diffusion web ui using nudenet and I like to say it works pretty nicely here's the repo url https://github.com/w-e-w/sd-webui-nudenet-nsfw-censor if you're interested

some key differences of my implementation

w-e-w commented 7 months ago

@poisenbery I'm fully aware that this is not a social media for self promotion

this post was merely to let the author that I have made something off his project similar to this post Onnx Post-Processing, NMS, TensoRT

I explained in post what I did different in my censoring filter

I believe my implantation has improved on his bare bones implementation of censoring which is just draw a black box on detected regions the issue with nudenet is that it's accuracy is not good enough to be used directly in that fashion

if the author wishes he can potentially reuse some of my code

bedapudi6788 commented 7 months ago

@w-e-w I am happy you shared. I will add the project in README. nice work!

w-e-w commented 7 months ago

@bedapudi6788 thank you it's an honor that you added it to readme