search

natethegreate / hent-AI

Automation of censor bar detection
MIT License
1.58k stars 145 forks source link

Issue with detecting censors in some images #5

Closed Zero3K closed 4 years ago

Zero3K commented 4 years ago

Some of the types of censors used by the images in https://u.teknik.io/4pTHC.zip aren't being detected by it.

natethegreate commented 4 years ago

I haven't looked into those images yet, but which censors in specific? As per the documentation, any type of censor that isn't a bar or mosaic is not supported.

Zero3K commented 4 years ago

The majority of the images use bars and mosaics. I think its the color of the bar(s) (or ones that have text on them) that causes an issue. There are a few that use words, etc. as censors.

natethegreate commented 4 years ago

Yeah bars that aren't near white or black won't get picked up, since it can get problematic training on uncommon colors. It is also pretty hard to perfectly mask anything under text.

We are currently working on expanding the dataset for further training, which will help the model detect much more.

Zero3K commented 4 years ago

Could you use the images that I provided to train it? And I think most of the bars in them are black/white.

natethegreate commented 4 years ago

The ai is missing these normal bars because it needs more training and a bigger dataset.

Images need to be annotated by hand in order to use them for training. If you want to try annotating them, feel free to join the Discord and we can teach you how to do it.