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

Nudenet Training #61

Closed ghost closed 3 years ago

ghost commented 3 years ago

Hello @bedapudi6788 . I am just learning how to teach and create artificial intelligence and I am afraid my problem will seem silly, but recently I came across your project Nudenet and I liked it. I would like to know if there is any way to teach it more? It is already well trained, but in some points it is too categorical and I would like to teach it a little bit more or teach it new content, and if possible, could you help me with that? Thank you in advance.

bedapudi6788 commented 3 years ago

For training/fine-tuning, you need to look at https://github.com/fizyr/keras-retinanet. In general, you will need more data tagged in that format and can finetune from one of the provided checkpoints.

ghost commented 3 years ago

Thank you for your quick answer, you helped me.