JanMarcelKezmann / TensorFlow-Advanced-Segmentation-Models

A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
Other
151 stars 48 forks source link

Typo in DeepLabV3+ with 16 output stride #16

Closed stdbreaks closed 2 years ago

stdbreaks commented 2 years ago

I suppose it must be output_layers = output_layers[:4]

https://github.com/JanMarcelKezmann/TensorFlow-Advanced-Segmentation-Models/blob/45edd430a9a6852cae474ebf3900284235a1710f/tensorflow_advanced_segmentation_models/models/DeepLabV3plus.py#L31

JanMarcelKezmann commented 2 years ago

Hi @stdbreaks ,

thank you for pointing it out, I have already pushed a fix.