notAI-tech / NudeNet

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

Training the classifier model #104

Open pranav318 opened 2 years ago

pranav318 commented 2 years ago

I am trying to train the nudenet classifier model. my doubt is that do we have to give only unsafe images to the model for training or we have to give both safe and unsafe content for it.

Thanks