Perfectfox256's picture
Create README.md
4b43272 verified
|
raw
history blame
334 Bytes

hotscreen-detection-models

NSFW bodyparts detection models to use with Hotscreen or to train your own.

You can use the script labelize-folder.py to labelize your dataset. You only have to install onnxruntime and cv2 in your python. Then edit the variables folder (containing your images) and model (the .onnx you want to use).