Bug fix in DeeplabV3

Author: JanMarcelKezmann

tensorflow_advanced_segmentation_models/__version__.py

@@ -1,3 +1,3 @@
-VERSION = (0, 4, 7)
+VERSION = (0, 4, 8)
 
 __version__ = ".".join(map(str, VERSION))

tensorflow_advanced_segmentation_models/models/DeepLabV3.py

@@ -28,7 +28,7 @@ def __init__(self, n_classes, base_model, output_layers, height=None, width=None
             self.dilations = [2 * rate for rate in dilations]
         elif self.output_stride == 16:
             self.final_upsampling2d = tf.keras.layers.UpSampling2D(size=16, interpolation="bilinear")
-            self.output_layers = self.output_layers[:4]
+            output_layers = output_layers[:4]
             self.dilations = dilations
         else:
             raise ValueError("'output_stride' must be one of (8, 16), got {}".format(self.output_stride))