Fix mask merge

Author: HaozhiQi

fcis/core/tester.py

@@ -129,8 +129,11 @@ def pred_eval(predictor, test_data, imdb, cfg, vis=False, thresh=1e-3, logger=No
                         all_masks[j][idx + delta] = cls_masks[keep, :]
                 else:
                     masks = masks[:, 1:, :, :]
+                    im_height = np.round(im_shapes[delta][0] / scales[delta]).astype('int')
+                    im_width = np.round(im_shapes[delta][1] / scales[delta]).astype('int')
+                    boxes = clip_boxes(boxes, (im_height, im_width))
                     result_mask, result_box = mask_voting(masks, boxes, scores, imdb.num_classes,
-                                                          max_per_image, im_shapes[delta][1], im_shapes[delta][0],
+                                                          max_per_image, im_width, im_height,
                                                           cfg.TEST.NMS, cfg.TEST.MASK_MERGE_THRESH,
                                                           cfg.BINARY_THRESH)
                     for j in xrange(1, imdb.num_classes):

fcis/demo.py

@@ -124,8 +124,12 @@ def main():
             masks = [all_masks[j] for j in range(1, num_classes)]
         else:
             masks = masks[0][:, 1:, :, :]
-            result_masks, result_dets = gpu_mask_voting(masks, boxes[0], scores[0], num_classes,
-                                                        100, im_shapes[0][1], im_shapes[0][0],
+            im_height = np.round(im_shapes[0][0] / scales[0]).astype('int')
+            im_width = np.round(im_shapes[0][1] / scales[0]).astype('int')
+            print (im_height, im_width)
+            boxes = clip_boxes(boxes[0], (im_height, im_width))
+            result_masks, result_dets = gpu_mask_voting(masks, boxes, scores[0], num_classes,
+                                                        100, im_width, im_height,
                                                         config.TEST.NMS, config.TEST.MASK_MERGE_THRESH,
                                                         config.BINARY_THRESH, ctx_id[0])