fix xcorr slow bugs

amandazw · Sep 23, 2019 · 71f4fd2 · 71f4fd2
1 parent 15ba130
commit 71f4fd2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysot/core/xcorr.py b/pysot/core/xcorr.py
@@ -17,8 +17,8 @@ def xcorr_slow(x, kernel):
     for i in range(batch):
         px = x[i]
         pk = kernel[i]
-        px = px.view(1, px.size()[0], px.size()[1], px.size()[2])
-        pk = pk.view(-1, px.size()[1], pk.size()[1], pk.size()[2])
+        px = px.view(1, -1, px.size()[1], px.size()[2])
+        pk = pk.view(1, -1, pk.size()[1], pk.size()[2])
         po = F.conv2d(px, pk)
         out.append(po)
     out = torch.cat(out, 0)