update

dontLoveBugs · May 4, 2020 · 1573ef8 · 1573ef8
1 parent 201b816
commit 1573ef8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/dorn/modules/encoders/SceneUnderstandingModule.py b/dorn/modules/encoders/SceneUnderstandingModule.py
@@ -67,11 +67,9 @@ def __init__(self, ord_num, size, kernel_size, pyramid=[6, 12, 18], batch_norm=F
         )
         self.concat_process = nn.Sequential(
             nn.Dropout2d(p=0.5),
-            nn.Conv2d(512 * 5, 2048, 1),
-            nn.ReLU(inplace=True),
+            conv_bn_relu(batch_norm, 512*5, 2048, kernel_size=1, padding=0),
             nn.Dropout2d(p=0.5),
-            nn.Conv2d(2048, ord_num * 2, 1),  # KITTI 142 NYU 136 In paper, K = 80 is best
-            # nn.UpsamplingBilinear2d(size=(385, 513))
+            nn.Conv2d(2048, ord_num * 2, 1)
         )
 
     def forward(self, x):