New class training and data about nsfw_model HOT 8 OPEN

@xaerincl are you planning on sharing your model?

I'd be interested in it.

from nsfw_model.

Ohh and also, i would love to see if you can share your test data so I can check how the model change in accuracy with the new class.

from nsfw_model.

(There's a bit tucked away in the "Extra Info" section on the main page about the dataset.)

from nsfw_model.

Hiya! I trained the model for a week, on my single GPU 1060

I'm not sure without experimenting how much data you'll need to bring in a new class. I'd love to hear how your training is going.

As for the training data, please see if it's still hosted here: #11

from nsfw_model.

A week for the inception model or for the mobilenet one?

btw your link works! any idea how can i download it using ubuntu terminal? im using paperspace to run my models.

from nsfw_model.

@GantMan btw the training data provided in your link is the same one you used to get 93% accuracy with the inceptionv3 net?

from nsfw_model.

@xaerincl yes!

from nsfw_model.

I just archieved 91% accuracy on the validation set with a simple mobilenet v2 (with no more layers than the original, i just changed the dimension of the last fc layer from 1000 to 5) in a full day of training with 11 complete epochs! (11cycles through all the data)

Whats was your final accuracy with the test set?

from nsfw_model.