search

notAI-tech / NudeNet

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

How to train model on custom dataset. #109

Open I-aashishM opened 2 years ago

I-aashishM commented 2 years ago

Hey @bedapudi6788

Can you please tell me how to train the model in our own dataset? I have also sent you an email regarding the same.

BlueBeret commented 10 months ago

Have you found the workaround? I need to fine tune this model with my own dataset ^_^