File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
segmentation_models_pytorch/decoders/deeplabv3 Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -43,7 +43,7 @@ class DeepLabV3(SegmentationModel):
43
43
- dropout (float): Dropout factor in [0, 1)
44
44
- activation (str): An activation function to apply "sigmoid"/"softmax"
45
45
(could be **None** to return logits)
46
- kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
46
+ kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
47
47
Keys with ``None`` values are pruned before passing.
48
48
49
49
Returns:
@@ -79,7 +79,7 @@ def __init__(
79
79
encoder_output_stride
80
80
)
81
81
)
82
-
82
+
83
83
self .encoder = get_encoder (
84
84
encoder_name ,
85
85
in_channels = in_channels ,
@@ -91,7 +91,7 @@ def __init__(
91
91
92
92
if upsampling is None :
93
93
if encoder_depth <= 3 :
94
- scale_factor = 2 ** encoder_depth
94
+ scale_factor = 2 ** encoder_depth
95
95
else :
96
96
scale_factor = encoder_output_stride
97
97
else :
@@ -153,7 +153,7 @@ class DeepLabV3Plus(SegmentationModel):
153
153
- dropout (float): Dropout factor in [0, 1)
154
154
- activation (str): An activation function to apply "sigmoid"/"softmax"
155
155
(could be **None** to return logits)
156
- kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
156
+ kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models.
157
157
Keys with ``None`` values are pruned before passing.
158
158
159
159
Returns:
You can’t perform that action at this time.
0 commit comments