From 2f0d20e49eb3d580f77236cc6d7cb9da4578e32a Mon Sep 17 00:00:00 2001 From: Alexandre Santos da Silva Jr Date: Fri, 16 Feb 2018 15:51:13 -0200 Subject: [PATCH] Updating model.py adding params to super() call Using inference.py call (with different parameters not documented, as --dataset is no longer available, for example) an error was encountered with the super() call, this update in the model.py function call solved the issue for me. --- model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/model.py b/model.py index b766bee..b4580df 100644 --- a/model.py +++ b/model.py @@ -8,11 +8,11 @@ class FCN8s(Network): def __init__(self, is_training=False, num_classes=151, input_size=[384, 384]): self.input_size = input_size - + self.x = tf.placeholder(dtype=tf.float32, shape=[None, None, 3]) self.img_tf, self.shape = preprocess(self.x, self.input_size, 'fcn-8s') - - super().__init__({'data': self.img_tf}, num_classes, is_training) + + super(FCN8s, self).__init__({'data': self.img_tf}, num_classes, is_training) def setup(self, is_training, num_classes): (self.feed('data')