search

notAI-tech / NudeNet

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

'classifier_checkpoint_path' #13

Closed lxknvlk closed 5 years ago

lxknvlk commented 5 years ago

It is unclear, what do i have to put in 'classifier_checkpoint_path' and 'detector_checkpoint_path'?

I have found this code in your repo

m = Detector('/Users/bedapudi/Desktop/inference_resnet50_csv_14.h5')

Can you share the inference_resnet50_csv_14.h5 file?

Are those the models from https://github.com/lxknvlk/NudeNet-models that i have to use in those paths?

bedapudi6788 commented 5 years ago

Just download the classifier_checkpoint from NueNet-models repo and pass it's path here.