In Tensorflow, wie die abgeflachten Indizes von tf.nn.max_pool_with_argmax zu entwirren?

Question

Ich treffe ein Problem: Nachdem ich die tf.nn.max_pool_with_argmax verwenden, erhalte ich die Indizes d argmax: A Tensor of type Targmax. 4-D. The flattened indices of the max values chosen for each output.

Wie der abgeflachte Indizes zurück auf die Koordinaten Liste in Tensorflow zu entwirren?

Vielen Dank.

Answer 1

Ich hatte das gleiche Problem heute und ich endete mit dieser Lösung:

def unravel_argmax(argmax, shape): 
    output_list = [] 
    output_list.append(argmax // (shape[2] * shape[3])) 
    output_list.append(argmax % (shape[2] * shape[3]) // shape[3]) 
    return tf.pack(output_list) 

Hier ist ein usage example in einem ipython Notebook (ich es verwenden, um die Pooling argmax Positionen meiner unpooling Methode zu übermitteln)