sd-webui-nsfw-filter icon indicating copy to clipboard operation
sd-webui-nsfw-filter copied to clipboard

algorithms of the filter might be too strict

Open sdwcn opened this issue 2 years ago • 8 comments

image i had tried with a few checkpoints, as shown above. Not sure if it's possible to have a weight option in order to adjust filtering rules?

sdwcn avatar Oct 22 '23 14:10 sdwcn

Sure, I'll add some thresholds and options for filtering (maybe, within 9 hours)

aria1th avatar Oct 22 '23 18:10 aria1th

Sure, I'll add some thresholds and options for filtering (maybe, within 9 hours)

how's it going?

sdwcn avatar Oct 25 '23 11:10 sdwcn

@sdwcn I tried some more setups, but seems like it has weird bias on white dress... The first plan was implementing labels to blur, but maybe it should have some very custom setups (such as thresholding by probability) Today, I'll release with some basic and custom setups, but I think model was roughly underfitted for border-images, resulting 'some' exposure to be censored

aria1th avatar Oct 26 '23 00:10 aria1th

noted, I'd like give a try,

sdwcn avatar Oct 27 '23 06:10 sdwcn

how shall I understand the diff between safe questionable & explicit?

levels_to_idx = {
        'safe': [1,3,4],
        'questionable': [1,3],
        'explicit': []
    }

according to tags in comments, ['drawings', 'hentai', 'neutral', 'porn', 'sexy'] does qustionable equals [hentai + porn]?

sdwcn avatar Oct 27 '23 06:10 sdwcn

Yes, safe = all (hentai porn sexy) questionable = hentai+porn explicit = none

aria1th avatar Oct 27 '23 09:10 aria1th

switched to "questionable", it still bias on white dresses, it sounds good if a probablity setting is enabled. cheers

sdwcn avatar Oct 30 '23 02:10 sdwcn

@sdwcn I can offer the probability option but its hyperparameter may require tremendous amout of test 😭

aria1th avatar Oct 30 '23 03:10 aria1th