fix: change default encoder for models

Author: okunator

cellseg_models_pytorch/models/cellpose/cellpose_unet.py

@@ -28,7 +28,7 @@ def __init__(
         out_channels: Tuple[int, ...] = (256, 128, 64, 32),
         layer_depths: Tuple[int, ...] = (4, 4, 4, 4),
         style_channels: int = 256,
-        enc_name: str = "resnet50",
+        enc_name: str = "efficientnet_b5",
         enc_pretrain: bool = True,
         enc_freeze: bool = False,
         enc_out_indices: Tuple[int, ...] = None,

cellseg_models_pytorch/models/cppnet/cppnet_unet.py

@@ -97,7 +97,7 @@ def __init__(
         depth: int = 4,
         out_channels: Tuple[int, ...] = (256, 128, 64, 32),
         style_channels: int = None,
-        enc_name: str = "resnet50",
+        enc_name: str = "efficientnet_b5",
         enc_pretrain: bool = True,
         enc_freeze: bool = False,
         enc_out_indices: Tuple[int, ...] = None,

cellseg_models_pytorch/models/hovernet/hovernet_unet.py

@@ -25,7 +25,7 @@ def __init__(
         depth: int = 4,
         out_channels: Tuple[int, ...] = (512, 256, 64, 64),
         style_channels: int = None,
-        enc_name: str = "resnet50",
+        enc_name: str = "efficientnet_b5",
         enc_pretrain: bool = True,
         enc_freeze: bool = False,
         enc_out_indices: Tuple[int, ...] = None,

cellseg_models_pytorch/models/stardist/stardist_unet.py

@@ -21,7 +21,7 @@ def __init__(
         depth: int = 4,
         out_channels: Tuple[int, ...] = (256, 128, 64, 32),
         style_channels: int = None,
-        enc_name: str = "resnet50",
+        enc_name: str = "efficientnet_b5",
         enc_pretrain: bool = True,
         enc_freeze: bool = False,
         enc_out_indices: Tuple[int, ...] = None,