File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed
segmentation_models_pytorch/decoders/dpt Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -54,10 +54,11 @@ class DPT(SegmentationModel):
54
54
**callable** and **None**. Default is **None**.
55
55
aux_params: Dictionary with parameters of the auxiliary output (classification head). Auxiliary output is build
56
56
on top of encoder if **aux_params** is not **None** (default). Supported params:
57
- - **classes** (*int*): A number of classes;
58
- - **pooling** (*str*): One of "max", "avg". Default is "avg";
59
- - **dropout** (*float*): Dropout factor in [0, 1);
60
- - **activation** (*str*): An activation function to apply "sigmoid"/"softmax" (could be **None** to return logits).
57
+
58
+ - **classes** (*int*): A number of classes;
59
+ - **pooling** (*str*): One of "max", "avg". Default is "avg";
60
+ - **dropout** (*float*): Dropout factor in [0, 1);
61
+ - **activation** (*str*): An activation function to apply "sigmoid"/"softmax" (could be **None** to return logits).
61
62
kwargs: Arguments passed to the encoder class ``__init__()`` function. Applies only to ``timm`` models. Keys with
62
63
``None`` values are pruned before passing. Specify ``dynamic_img_size=True`` to allow the model to handle images of different sizes.
63
64
You can’t perform that action at this time.
0 commit comments