From 62f42d5113e128d192df6af47a38a0c9cb148c6c Mon Sep 17 00:00:00 2001 From: seoyoonbaek Date: Sat, 1 May 2021 18:37:44 +0900 Subject: [PATCH] =?UTF-8?q?#18=20=EB=AA=A8=EB=8D=B8=EC=B6=94=EA=B0=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- model/MANet(effib7).py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 model/MANet(effib7).py diff --git a/model/MANet(effib7).py b/model/MANet(effib7).py new file mode 100644 index 0000000..ceb5e89 --- /dev/null +++ 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 \ No newline at end of file