INNER CODE UNIT · Python

init

introlab/rtabmap · corelib/src/python/rtabmap_netvlad.py:30

def init(descriptorDim):
    print("NetVLAD python init()")
    global image_batch
    global net_out
    global saver
    global sess
    global dim
    
    dim = descriptorDim

    tf.compat.v1.disable_eager_execution()
    tf.compat.v1.reset_default_graph()

    image_batch = tf.compat.v1.placeholder(
        dtype=tf.float32, shape=[None, None, None, 3])

    net_out = nets.vgg16NetvladPca(image_batch)
    saver = tf.compat.v1.train.Saver()

View source record →