Skip to content

Commit f814adc

Browse files
committed
Rename for FPN
1 parent 59cc7de commit f814adc

File tree

1 file changed

+3
-3
lines changed
  • segmentation_models_pytorch/decoders/fpn

1 file changed

+3
-3
lines changed

segmentation_models_pytorch/decoders/fpn/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ class FPN(SegmentationModel):
2828
decoder_merge_policy: Determines how to merge pyramid features inside FPN. Available options are **add**
2929
and **cat**
3030
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
3232
**"nearest"**, **"bilinear"**, **"bicubic"**, **"area"**, **"nearest-exact"**. Default is **"nearest"**.
3333
in_channels: A number of input channels for the model, default is 3 (RGB images)
3434
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__(
6464
decoder_segmentation_channels: int = 128,
6565
decoder_merge_policy: str = "add",
6666
decoder_dropout: float = 0.2,
67-
decoder_interpolation_mode: str = "nearest",
67+
decoder_interpolation: str = "nearest",
6868
in_channels: int = 3,
6969
classes: int = 1,
7070
activation: Optional[str] = None,
@@ -95,7 +95,7 @@ def __init__(
9595
segmentation_channels=decoder_segmentation_channels,
9696
dropout=decoder_dropout,
9797
merge_policy=decoder_merge_policy,
98-
interpolation_mode=decoder_interpolation_mode,
98+
interpolation_mode=decoder_interpolation,
9999
)
100100

101101
self.segmentation_head = SegmentationHead(

0 commit comments

Comments
 (0)