#18 모델추가

bcaitech1 · May 1, 2021 · 62f42d5 · 62f42d5
1 parent 90d0883
commit 62f42d5
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/model/MANet(effib7).py b/model/MANet(effib7).py
@@ -0,0 +1,17 @@
+import torch.nn as nn
+import segmentation_models_pytorch as smp
+
+class MANet_efficientNet_b7(nn.Module):
+    def __init__(self, in_channels = 3, classes = 12):
+        super(MANet_efficientNet_b7, self).__init__()
+        self.backbone = smp.MAnet(
+            encoder_name="efficientnet-b7",     
+            encoder_weights="imagenet",
+            in_channels=3,               
+            classes=12
+        )
+
+    def forward(self, x):
+        output = self.backbone(x)
+
+        return output