Skip to content

Commit

Permalink
modify docstring for upsampling of DeepLabV3Plus
Browse files Browse the repository at this point in the history
  • Loading branch information
munehiro-k committed Nov 24, 2024
1 parent 3179751 commit 2efd974
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion segmentation_models_pytorch/decoders/deeplabv3/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,8 @@ class DeepLabV3Plus(SegmentationModel):
Available options are **"sigmoid"**, **"softmax"**, **"logsoftmax"**, **"tanh"**, **"identity"**,
**callable** and **None**.
Default is **None**
upsampling: Final upsampling factor. Default is 4 to preserve input-output spatial shape identity
upsampling: Final upsampling factor. Default is 4 to preserve input-output spatial shape identity. In case
**encoder_depth** and **encoder_output_stride** are 3 and 16 resp., set **upsampling** to 2 to preserve.
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
on top of encoder if **aux_params** is not **None** (default). Supported params:
- classes (int): A number of classes
Expand Down

0 comments on commit 2efd974

Please sign in to comment.