Exemplo n.º 1
0
 def py_load_pickle(path):
     depthmap, targets = pickle.load(open(path.numpy(), "rb"))
     depthmap = preprocess_depthmap(depthmap)
     depthmap = tf.image.resize(depthmap,
                                (image_target_height, image_target_width))
     targets = preprocess_targets(targets, targets_indices)
     return depthmap, targets
Exemplo n.º 2
0
 def py_load_pickle(path, max_value):
     depthmap, targets = pickle.load(open(path.numpy(), "rb"))
     depthmap = preprocess_depthmap(depthmap)
     depthmap = depthmap / max_value
     depthmap = tf.image.resize(depthmap, (CONFIG.IMAGE_TARGET_HEIGHT, CONFIG.IMAGE_TARGET_WIDTH))
     targets = preprocess_targets(targets, targets_indices)
     return depthmap, targets
Exemplo n.º 3
0
 def py_load_pickle(path):
     depthmap, targets = pickle.load(open(path.numpy(), "rb"))
     depthmap = preprocess_depthmap(depthmap)
     targets = preprocess_targets(targets, targets_indices)
     return depthmap, targets
Exemplo n.º 4
0
 def find_max(path):
     depthmap, targets = pickle.load(open(path.numpy(), "rb"))
     depthmap = preprocess_depthmap(depthmap)
     depthmax = tf.math.reduce_max(depthmap)
     return depthmax