notAI-tech / NudeNet

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

Is it possible to train the nudeclassifier model on additional examples? #35

Closed aayushbhaskar closed 4 years ago

aayushbhaskar commented 4 years ago

I want to train the model on some additional images which contain subtle nudity. Also, the nudedetector model starts employing about 24 cores of the server after testing on a few images which causes a heavy load on the server. Is there a workaround for this problem.

bedapudi6788 commented 4 years ago

I want to train the model on some additional images which contain subtle nudity.

Search for "finetune image classification model keras", and you will find a number of tutorials.

Also, the nudedetector model starts employing about 24 cores of the server after testing on a few images which causes a heavy load on the server.

It is impossible to debug without relevant information.

bedapudi6788 commented 4 years ago

@aayushbhaskar closing the issue for now. Feel free to re-open if needed.