imageimage-processingtensorflowtagging

Tensorflow Auto Tagging


I want to know how to make an auto-tagging for images.

I have tried using Tensorflow and pre-trained the models several times. For start, it was quite good for classification.

But now, I need to do auto-tagging.

Using tensorflow the prediction sum result will be always 1.

For example something like this :

xxx.jpg prediction result :

  • Cat = 0.822
  • Dog = 0.177
  • Deer = 0.001

The sum will be always 1.

What I wanted is something like this :

xxx.jpg prediction result :

  • Cat = 0.901
  • Dog = 0.811
  • Deer = 0.991

Because there might be Cat, Dog and Deer in the same picture in xxx.jpg (just like Clarifai did.)

I wonder what is the basic concept to achieved that?

Thank you.


Solution

  • Take a look at the last layer you made. As you say the sum of your predictions is always one, this sounds like you applied softmax (https://en.wikipedia.org/wiki/Softmax_function). If you remove this, you get activations for every object.

    Please let me know if this helped you!