notAI-tech / NudeNet

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

Centering censorship area around specific points #145

Open CortesJoan opened 2 months ago

CortesJoan commented 2 months ago

Currently, when using the code to make the censorship area smaller, some parts are not being censored correctly. This is because the censorship does not start with the desired point (e.g., nipples) as the center.

Example Image:

Example Image:

image

As you can see in the example image, the censorship area is not centered around the nipples, resulting in parts of the nipples being visible.

Desired Behavior:

The censorship area should be centered around specific points (e.g., nipples) when making the censorship area smaller. This would ensure that the desired regions are properly censored, even with a smaller censorship area.

bedapudi6788 commented 1 day ago

@CortesJoan this requires a newer different type of model which I am working on. the current models will only detect the full breast/ chest area, so it cant be done with the current model as it is.

WIll push the new one in a week or so