Skip to content

Commit

Permalink
Update docs 2
Browse files Browse the repository at this point in the history
  • Loading branch information
qubvel committed Aug 7, 2019
1 parent 9a5280b commit 8fbf9be
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 3 deletions.
3 changes: 0 additions & 3 deletions segmentation_models/models/fpn.py
Original file line number Diff line number Diff line change
Expand Up @@ -195,8 +195,6 @@ def FPN(
input_shape: shape of input data/image ``(H, W, C)``, in general
case you do not need to set ``H`` and ``W`` shapes, just pass ``(None, None, C)`` to make your model be
able to process images af any size, but ``H`` and ``W`` of input images should be divisible by factor ``32``.
input_tensor: optional Keras tensor (i.e. output of `layers.Input()`) to use as image input for the model
(works only if ``encoder_weights`` is ``None``).
classes: a number of classes for output (output shape - ``(h, w, classes)``).
weights: optional, path to model weights.
activation: name of one of ``keras.activations`` for last model layer (e.g. ``sigmoid``, ``softmax``, ``linear``).
Expand All @@ -210,7 +208,6 @@ def FPN(
is used.
pyramid_aggregation: one of 'sum' or 'concat'. The way to aggregate pyramid blocks.
pyramid_dropout: spatial dropout rate for feature pyramid in range (0, 1).
final_interpolation: interpolation type for upsampling layers, on of ``nearest``, ``bilinear``.
Returns:
``keras.models.Model``: **FPN**
Expand Down
1 change: 1 addition & 0 deletions segmentation_models/models/linknet.py
Original file line number Diff line number Diff line change
Expand Up @@ -210,6 +210,7 @@ def Linknet(
classes: a number of classes for output (output shape - ``(h, w, classes)``).
activation: name of one of ``keras.activations`` for last model layer
(e.g. ``sigmoid``, ``softmax``, ``linear``).
weights: optional, path to model weights.
encoder_weights: one of ``None`` (random initialization), ``imagenet`` (pre-training on ImageNet).
encoder_freeze: if ``True`` set all layers of encoder (backbone model) as non-trainable.
encoder_features: a list of layer numbers or names starting from top of the model.
Expand Down

0 comments on commit 8fbf9be

Please sign in to comment.