fixes NRCan#293 NRCan#246

add tests for optimizer instantiation in test_optimizers.py adapt our unet models (models/unet.py) to expect same parameter names as smp models
remtav · Mar 24, 2022 · 6275fb0 · 6275fb0
1 parent bd0c88a
commit 6275fb0
Show file tree

Hide file tree

Showing 10 changed files with 65 additions and 0 deletions.
diff --git a/config/model/deeplabv3_smp.yaml b/config/model/deeplabv3_smp.yaml
@@ -0,0 +1,7 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.DeepLabV3
+  encoder_name: resnet101
+  encoder_weights: imagenet
+
+
diff --git a/config/model/deeplabv3plus_smp.yaml b/config/model/deeplabv3plus_smp.yaml
@@ -0,0 +1,7 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.DeepLabV3Plus
+  encoder_name: resnext50_32x4d
+  encoder_weights: imagenet
+  classes:
+  in_channels:
diff --git a/config/model/manet_smp.yaml b/config/model/manet_smp.yaml
@@ -0,0 +1,5 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.MAnet
+  encoder_name: resnext101_32x8d
+  encoder_weights: imagenet
diff --git a/config/model/pan_smp.yaml b/config/model/pan_smp.yaml
@@ -0,0 +1,5 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.PAN
+  encoder_name: se_resnext101_32x4d
+  encoder_weights: imagenet
diff --git a/config/model/spacenet_unet_baseline_smp.yaml b/config/model/spacenet_unet_baseline_smp.yaml
@@ -0,0 +1,6 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.Unet
+  encoder_name: vgg11
+  encoder_weights: imagenet
+
diff --git a/config/model/spacenet_unet_efficientnetb5_smp.yaml b/config/model/spacenet_unet_efficientnetb5_smp.yaml
@@ -0,0 +1,6 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.Unet
+  encoder_name: efficientnet-b5
+  encoder_weights: imagenet
+
diff --git a/config/model/spacenet_unet_senet154_smp.yaml b/config/model/spacenet_unet_senet154_smp.yaml
@@ -0,0 +1,5 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.Unet
+  encoder_name: senet154
+  encoder_weights: imagenet
diff --git a/config/model/unet_plusplus_smp.yaml b/config/model/unet_plusplus_smp.yaml
@@ -0,0 +1,10 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.UnetPlusPlus
+  encoder_name: se_resnext50_32x4d
+  encoder_depth: 4
+  decoder_channels: [256, 128, 64, 32]
+  decoder_attention_type: scse
+  encoder_weights: imagenet
+
+
diff --git a/config/model/unet_resnext101_smp.yaml b/config/model/unet_resnext101_smp.yaml
@@ -0,0 +1,7 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.Unet
+  encoder_name: resnext101_32x8d
+  encoder_depth: 4
+  decoder_channels: [ 256, 128, 64, 32 ]
+  encoder_weights: imagenet
diff --git a/config/model/unet_smp.yaml b/config/model/unet_smp.yaml
@@ -0,0 +1,7 @@
+# @package _global_
+model:
+  _target_: segmentation_models_pytorch.Unet
+  encoder_name: resnext50_32x4d
+  encoder_depth: 5
+  encoder_weights: imagenet
+