modify

JiaRenChang · JiaRenChang · commit a80dee7425f1 · 2018-04-10T12:30:41.000+08:00
diff --git a/finetune.py b/finetune.py
@@ -131,7 +131,7 @@ def test(imgL,imgR,disp_true):
         correct = (disp_true[index[0][:], index[1][:], index[2][:]] < 3)+(disp_true[index[0][:], index[1][:], index[2][:]] < true_disp[index[0][:], index[1][:], index[2][:]]*0.05)      
         torch.cuda.empty_cache()
 
-        return float(torch.sum(correct))/float(len(index[0]))
+        return 1-(float(torch.sum(correct))/float(len(index[0])))
 
 def adjust_learning_rate(optimizer, epoch):
     if epoch <= 200:
diff --git a/models/submodule.py b/models/submodule.py
@@ -9,7 +9,7 @@
 
 def convbn(in_planes, out_planes, kernel_size, stride, pad, dilation):
 
-    return nn.Sequential(nn.Conv2d(in_planes, out_planes, kernel_size=kernel_size, stride=stride, padding=dilation, dilation = dilation, bias=False),
+    return nn.Sequential(nn.Conv2d(in_planes, out_planes, kernel_size=kernel_size, stride=stride, padding=dilation if dilation > 1 else pad, dilation = dilation, bias=False),
                          nn.BatchNorm2d(out_planes))
 
 
