updated __init__ functions

Author: JanMarcelKezmann

__init__.py

@@ -1 +1 @@
-from .tf_segmentation_models import *
+from .tensorflow_advanced_segmentation_models import *

tensorflow_advanced_segmentation_models/__init__.py

@@ -1 +1,14 @@
+from . import base
+from . import losses
+from . import metrics
 
+from .backbones.tf_backbones import create_base_model
+from .models.FCN import FCN
+from .models.UNet import UNet
+from .models.OCNet import OCNet
+from .models.FPNet import FPNet
+from .models.DANet import DANet
+from .models.PSPNet import PSPNet
+from .models.DeepLab import DeepLab
+from .models.DeepLabV3 import DeepLabV3
+from .models.DeepLabV3plus import DeepLabV3plus

tensorflow_advanced_segmentation_models/models/DeepLabV3plus.py

@@ -1,6 +1,6 @@
 import tensorflow as tf
 
-from ._custom_layers_and_blocks import ConvolutionBnActivation, AtrousSeparableConvolutionBnReLU, AtrousSpatialPyramidPooling
+from ._custom_layers_and_blocks import ConvolutionBnActivation, AtrousSeparableConvolutionBnReLU, AtrousSpatialPyramidPoolingV3
 from ..backbones.tf_backbones import create_base_model
 
 class DeepLabV3plus(tf.keras.Model):