add point finetune: fix bug of valid_boxes

Author: Mark-ZhouWX

official/cv/segment-anything/segment_anything/modeling/loss.py

@@ -32,7 +32,7 @@ def __init__(self, focal_factor=20.0, dice_factor=1.0, mse_factor=1.0, mask_thre
         self.dice_loss = DiceLoss(reduction='none')
         self.mse_loss = nn.MSELoss(reduction='none')
 
-    def construct(self, pred_mask, pred_iou, gt_mask, valid_boxes=None):
+    def construct(self, pred_mask, pred_iou, gt_mask, valid_boxes):
         """
         get loss, remove dynamic shape assisted with valid_boxes
         Args:

official/cv/segment-anything/segment_anything/utils/model_wrapper.py

@@ -224,7 +224,7 @@ def forward_point(image, points=None, boxes=None, masks=None,
                           multimask_output=False, output_best_mask=True, return_low_res_mask=True):
             _pred_mask, _pred_iou, _low_res_mask = net(image, points=points, boxes=boxes, masks=masks,
                                                        multimask_output=multimask_output, output_best_mask=output_best_mask, return_low_res_mask=return_low_res_mask)
-            _loss = loss_fn(_pred_mask, _pred_iou, gt_mask=gt_mask, valid_boxes=valid_boxes)
+            _loss = loss_fn(_pred_mask, _pred_iou, gt_mask, valid_boxes)
             return loss_scaler.scale(_loss[0]), (_pred_mask, _pred_iou, _low_res_mask)
 
         def _train_fn(*data_element):