added focal loss

YangXS · Sep 21, 2017 · c69573f · c69573f
1 parent c726bc8
commit c69573f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/cfgs/res101.yml b/cfgs/res101.yml
@@ -14,3 +14,5 @@ TRAIN:
 TEST:
   HAS_RPN: True
 POOLING_SIZE: 7
+POOLING_MODE: align
+CROP_RESIZE_WITH_MAX_POOL: False
diff --git a/lib/model/faster_rcnn/faster_rcnn_cascade.py b/lib/model/faster_rcnn/faster_rcnn_cascade.py
@@ -147,7 +147,12 @@ def forward(self, im_data, im_info, gt_boxes, num_boxes):
             self.fg_cnt = torch.sum(label.data.ne(0))
             self.bg_cnt = label.data.numel() - self.fg_cnt
 
-            self.RCNN_loss_cls = F.cross_entropy(cls_score, label)
+            # focal loss
+            weights = (1 - cls_prob) ** 2
+            focal_prob = torch.log(cls_prob) * weights.detach()
+            self.RCNN_loss_cls = F.nll_loss(focal_prob, label)
+
+            # self.RCNN_loss_cls = F.cross_entropy(cls_score, label, weights.detach())
 
             # bounding box regression L1 loss
             self.RCNN_loss_bbox = _smooth_l1_loss(bbox_pred, rois_target, rois_inside_ws, rois_outside_ws)