himanshuch8055 commited on
Commit
e8ee68b
·
1 Parent(s): 358d1eb

Replace UNet with DeepLabV3Plus in model loading for improved segmentation performance

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -294,7 +294,7 @@ def load_model(model_name):
294
  activation=None
295
  )
296
  elif config["architecture"] == "DeepLabV3Plus":
297
- model = smp.Unet(
298
  encoder_name=config["encoder"],
299
  encoder_weights="imagenet",
300
  # decoder_channels=(256, 128, 64, 32, 16), # Not used in DeepLabV3Plus
 
294
  activation=None
295
  )
296
  elif config["architecture"] == "DeepLabV3Plus":
297
+ model = smp.DeepLabV3Plus(
298
  encoder_name=config["encoder"],
299
  encoder_weights="imagenet",
300
  # decoder_channels=(256, 128, 64, 32, 16), # Not used in DeepLabV3Plus