Spaces:
Sleeping
Sleeping
Commit ·
e8ee68b
1
Parent(s): 358d1eb
Replace UNet with DeepLabV3Plus in model loading for improved segmentation performance
Browse files
app.py
CHANGED
|
@@ -294,7 +294,7 @@ def load_model(model_name):
|
|
| 294 |
activation=None
|
| 295 |
)
|
| 296 |
elif config["architecture"] == "DeepLabV3Plus":
|
| 297 |
-
model = smp.
|
| 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
|