Merge pull request STVIR#44 from wwdguu/upc

fix UpChannelRPN
amandazw · Jun 12, 2019 · 52ef3aa · 52ef3aa
2 parents c138ff8 + 6b045f4
commit 52ef3aa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysot/models/head/rpn.py b/pysot/models/head/rpn.py
@@ -32,9 +32,9 @@ def __init__(self, anchor_num=5, feature_in=256):
                 feature_in * loc_output, kernel_size=3)
 
         self.search_cls_conv = nn.Conv2d(feature_in, 
-                feature_in * cls_output, kernel_size=3)
+                feature_in, kernel_size=3)
         self.search_loc_conv = nn.Conv2d(feature_in, 
-                feature_in * loc_output, kernel_size=3)
+                feature_in, kernel_size=3)
 
         self.loc_adjust = nn.Conv2d(loc_output, loc_output, kernel_size=1)