search

mdietrichstein / tensorflow-open_nsfw

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

when I convert with Caffe to TensorFlow method， I got wrong score 0.95353949 #7

Closed hdmjdp closed 6 years ago

hdmjdp commented 6 years ago

And I have found the difference is between model.py. Can you show me your convert methods in detail? thanks

mdietrichstein commented 6 years ago

I have used caffe-tensorflow to extract the model weights. The model (model.py) was created by looking at the original implementation and manually porting it over to tensorflow.

There were some adaptions necessary regarding the padding layers since caffe and tensorflow have different approaches there.