diff --git a/lib/model/rpn/rpn.py b/lib/model/rpn/rpn.py index 29bdaadb3..15012e143 100644 --- a/lib/model/rpn/rpn.py +++ b/lib/model/rpn/rpn.py @@ -27,7 +27,7 @@ def __init__(self, din): # define the convrelu layers processing input feature map self.RPN_Conv = nn.Conv2d(self.din, 512, 3, 1, 1, bias=True) - # define bg/fg classifcation score layer + # define bg/fg classification score layer self.nc_score_out = len(self.anchor_scales) * len(self.anchor_ratios) * 2 # 2(bg/fg) * 9 (anchors) self.RPN_cls_score = nn.Conv2d(512, self.nc_score_out, 1, 1, 0)