From 6ce121f861e5860888112f06361363e7c5a6bd02 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 14 Dec 2022 17:36:30 +0000 Subject: [PATCH] DeepLabV3+: have argument for number of channels --- aimodel/src/deeplabv3_plus_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/deeplabv3_plus_test.py b/aimodel/src/deeplabv3_plus_test.py index b0a7d03..0644e6c 100755 --- a/aimodel/src/deeplabv3_plus_test.py +++ b/aimodel/src/deeplabv3_plus_test.py @@ -118,8 +118,8 @@ def DilatedSpatialPyramidPooling(dspp_input): return output -def DeeplabV3Plus(image_size, num_classes): - model_input = tf.keras.Input(shape=(image_size, image_size, 3)) +def DeeplabV3Plus(image_size, num_classes, num_channels=3): + model_input = tf.keras.Input(shape=(image_size, image_size, num_channels)) resnet50 = tf.keras.applications.ResNet50( weights="imagenet", include_top=False, input_tensor=model_input )