hourglass_tiny.py 文件源码

def _argmax(self, tensor):
        """ ArgMax
        Args:
            tensor  : 2D - Tensor (Height x Width : 64x64 )
        Returns:
            arg     : Tuple of max position
        """
        resh = tf.reshape(tensor, [-1])
        argmax = tf.arg_max(resh, 0)
        return (argmax // tensor.get_shape().as_list()[0], argmax % tensor.get_shape().as_list()[0])