Changed no. of channels ultralytics#18

manole-alexandru · Apr 20, 2023 · f6e14f9 · f6e14f9
1 parent 4d49801
commit f6e14f9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/models/common.py b/models/common.py
@@ -852,10 +852,10 @@ class Seg(nn.Module):
     def __init__(self, in_channels):
         super().__init__()
         print('SEG in channels: ', in_channels)
-        self.cv1 = Conv(in_channels, 32, k=3)
+        self.cv1 = Conv(in_channels, 64, k=3)
         self.upsample = nn.Upsample(scale_factor=2, mode='nearest')
-        self.cv2 = Conv(32, 64, k=3)
-        self.cv3 = Conv(64, 64, k=3)
+        self.cv2 = Conv(64, 128, k=3)
+        self.cv3 = Conv(128, 64, k=3)
         self.cv4 = Conv(64, 1, act=False)
         self.relu = nn.ReLU()