search

GantMan / nsfw_model

Keras model of NSFW detector
Other
1.8k stars 279 forks source link

Training with mobilenet? #20

Closed onexuan closed 5 years ago

onexuan commented 5 years ago

Training with mobilenet?

GantMan commented 5 years ago

Sure, that can be a thing. Is this something you'd like to take on? Or are you asking for code on how to do this?

onexuan commented 5 years ago

I hope you support it. It can be used for mobile phone image classification, which will be great

GantMan commented 5 years ago

Gotcha, so you're looking for a smaller/faster model. I'll keep this ticket open, and help whoever wants to do it. If no one, I may just take it on myself.

onexuan commented 5 years ago

it is glad to hear that!

GantMan commented 5 years ago

Currently, over 80% accurate, I'll tune it over the weekend and update you.

onexuan commented 5 years ago

Great work, I know you train it with Tensorflow, I wish I could train like you. I hope it support mxnet for me, I don't how to train, but i know use with mxnet in mobile phone!

GantMan commented 5 years ago

90% accurate, gonna fine-tune

onexuan commented 5 years ago

perfect! waiting...

GantMan commented 5 years ago

Here: image

Download Keras Model:

I'll keep trying to get it more accurate and will update the main README.md when it's ready. https://s3.amazonaws.com/ir_public/nsfwjscdn/nsfw_mobilenet2.224x224.h5

onexuan commented 5 years ago

it is possible to also publish 112x112 at the same time! Thank you for your great work, I will try to covert it to other model used in android