search

mdietrichstein / tensorflow-open_nsfw

Tensorflow Implementation of Yahoo's Open NSFW Model
Other
430 stars 136 forks source link

improve the model #3

Closed mohitrgiit closed 6 years ago

mohitrgiit commented 7 years ago

any idea how can i improve the model and tune it with my dataset

mdietrichstein commented 6 years ago

The model is a standard resnet model that you should be able to finetune like any other model with tensorflow. The process is a bit involved, but you can find a nice tensorflow finetuning guide at https://kratzert.github.io/2017/02/24/finetuning-alexnet-with-tensorflow.html