@@ -28,7 +28,7 @@ class FPN(SegmentationModel):
28
28
decoder_merge_policy: Determines how to merge pyramid features inside FPN. Available options are **add**
29
29
and **cat**
30
30
decoder_dropout: Spatial dropout rate in range (0, 1) for feature pyramid in FPN_
31
- decoder_interpolation_mode : Interpolation mode used in decoder of the model. Available options are
31
+ decoder_interpolation : Interpolation mode used in decoder of the model. Available options are
32
32
**"nearest"**, **"bilinear"**, **"bicubic"**, **"area"**, **"nearest-exact"**. Default is **"nearest"**.
33
33
in_channels: A number of input channels for the model, default is 3 (RGB images)
34
34
classes: A number of classes for output mask (or you can think as a number of channels of output mask)
@@ -64,7 +64,7 @@ def __init__(
64
64
decoder_segmentation_channels : int = 128 ,
65
65
decoder_merge_policy : str = "add" ,
66
66
decoder_dropout : float = 0.2 ,
67
- decoder_interpolation_mode : str = "nearest" ,
67
+ decoder_interpolation : str = "nearest" ,
68
68
in_channels : int = 3 ,
69
69
classes : int = 1 ,
70
70
activation : Optional [str ] = None ,
@@ -95,7 +95,7 @@ def __init__(
95
95
segmentation_channels = decoder_segmentation_channels ,
96
96
dropout = decoder_dropout ,
97
97
merge_policy = decoder_merge_policy ,
98
- interpolation_mode = decoder_interpolation_mode ,
98
+ interpolation_mode = decoder_interpolation ,
99
99
)
100
100
101
101
self .segmentation_head = SegmentationHead (
0 commit comments