Commit 46d12bdf authored by Fernando Perez-Garcia's avatar Fernando Perez-Garcia

Use default config at inference for BN

parent afc14883
......@@ -453,7 +453,7 @@ class SegmentationApplicationVessels(BaseApplication):
# classification probabilities or argmax classification labels
data_dict = switch_sampler(for_training=False)
image = tf.cast(data_dict['image'], tf.float32)
net_args = {'is_training': True, # self.is_training, (Reuben modif)
net_args = {'is_training': self.is_training, # True # (Reuben modif)
'keep_prob': self.net_param.keep_prob}
net_out = self.net(image, **net_args)
......
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