From ce427a92ffcdc9f36245f9703ab0be48c2a2ab86 Mon Sep 17 00:00:00 2001 From: Erik Date: Mon, 6 Jan 2020 11:49:03 +0100 Subject: [PATCH] fixed typos and non-PEP8 whitespace --- lib/model/faster_rcnn/faster_rcnn.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/lib/model/faster_rcnn/faster_rcnn.py b/lib/model/faster_rcnn/faster_rcnn.py index 649616260..39bec118b 100644 --- a/lib/model/faster_rcnn/faster_rcnn.py +++ b/lib/model/faster_rcnn/faster_rcnn.py @@ -53,7 +53,7 @@ def forward(self, im_data, im_info, gt_boxes, num_boxes): # feed base feature map tp RPN to obtain rois rois, rpn_loss_cls, rpn_loss_bbox = self.RCNN_rpn(base_feat, im_info, gt_boxes, num_boxes) - # if it is training phrase, then use ground trubut bboxes for refining + # if it is training phrase, then use ground truth bboxes for refining if self.training: roi_data = self.RCNN_proposal_target(rois, gt_boxes, num_boxes) rois, rois_label, rois_target, rois_inside_ws, rois_outside_ws = roi_data @@ -103,7 +103,6 @@ def forward(self, im_data, im_info, gt_boxes, num_boxes): # bounding box regression L1 loss RCNN_loss_bbox = _smooth_l1_loss(bbox_pred, rois_target, rois_inside_ws, rois_outside_ws) - cls_prob = cls_prob.view(batch_size, rois.size(1), -1) bbox_pred = bbox_pred.view(batch_size, rois.size(1), -1) @@ -112,7 +111,7 @@ def forward(self, im_data, im_info, gt_boxes, num_boxes): def _init_weights(self): def normal_init(m, mean, stddev, truncated=False): """ - weight initalizer: truncated normal and random normal. + weight initializer: truncated normal and random normal. """ # x is a parameter if truncated: