Skip to content

GitLab

Unverified Commit a9eaf98e authored by lkskstlr's avatar lkskstlr Committed by GitHub
Browse files

struct2depth inference.py bug fix (#7342)

parent 5e34a97b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/struct2depth/model.py
View file @ a9eaf98e
...@@ -761,6 +761,7 @@ class Model(object): ...@@ -761,6 +761,7 @@ class Model(object):
input_image = tf.placeholder( input_image = tf.placeholder(
tf.float32, [self.batch_size, self.img_height, self.img_width, 3], tf.float32, [self.batch_size, self.img_height, self.img_width, 3],
name='raw_input') name='raw_input')
self.input_image = input_image
if self.imagenet_norm: if self.imagenet_norm:
input_image = (input_image - reader.IMAGENET_MEAN) / reader.IMAGENET_SD input_image = (input_image - reader.IMAGENET_MEAN) / reader.IMAGENET_SD
est_disp, _ = nets.disp_net(architecture=self.architecture, est_disp, _ = nets.disp_net(architecture=self.architecture,
...@@ -769,7 +770,6 @@ class Model(object): ...@@ -769,7 +770,6 @@ class Model(object):
weight_reg=self.weight_reg, weight_reg=self.weight_reg,
is_training=True) is_training=True)
est_depth = 1.0 / est_disp[0] est_depth = 1.0 / est_disp[0]
self.input_image = input_image
self.est_depth = est_depth self.est_depth = est_depth
def build_egomotion_test_graph(self): def build_egomotion_test_graph(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment