Add clip\clamp activation (qubvel#518)

Author: Vozf

segmentation_models_pytorch/base/modules.py

@@ -73,6 +73,15 @@ def forward(self, x):
         return torch.argmax(x, dim=self.dim)
 
 
+class Clamp(nn.Module):
+    def __init__(self, min=0, max=1):
+        super().__init__()
+        self.min, self.max = min, max
+
+    def forward(self, x):
+        return torch.clamp(x, self.min, self.max)
+
+
 class Activation(nn.Module):
 
     def __init__(self, name, **params):
@@ -95,6 +104,8 @@ def __init__(self, name, **params):
             self.activation = ArgMax(**params)
         elif name == 'argmax2d':
             self.activation = ArgMax(dim=1, **params)
+        elif name == 'clamp':
+            self.activation = Clamp(**params)
         elif callable(name):
             self.activation = name(**params)
         else: