AUTOMATIC1111 / stable-diffusion-webui-nsfw-censor

stable-diffusion-webui-nsfw-censor
116 stars 47 forks source link

Inaccurate NSFW detection results #4

Open Marx2014 opened 1 year ago

Marx2014 commented 1 year ago

Why is it detecting NSFW even when I upload a some image? image is: bug_handle

my code:

def check_image(input_image):
    test_image = Image.open(input_image)
    x_image = np.array(test_image)
    _, has_nsfw_concept = check_safety(x_image)
    return has_nsfw_concept

print(check_image("bug.png"))

print result: Potential NSFW content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed. [True]