Update resnet101_asp_oc.py

update the asp_oc
openseg-group · Jun 11, 2019 · 953469f · 953469f
1 parent 6a3a0b9
commit 953469f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/network/resnet101_asp_oc.py b/network/resnet101_asp_oc.py
@@ -64,7 +64,9 @@ def __init__(self, block, layers, num_classes):
 
         # extra added layers
         self.context = nn.Sequential(
-                ASP_OC_Module(2048, 512),
+                nn.Conv2d(2048, 512, kernel_size=3, stride=1, padding=1),
+                InPlaceABNSync(512),
+                ASP_OC_Module(512, 512)
                 )
         self.cls = nn.Conv2d(512, num_classes, kernel_size=1, stride=1, padding=0, bias=True)
         self.dsn = nn.Sequential(