93073ef18313015f2f45d9b54106ed98c8dc2579,mmdet/core/post_processing/merge_augs.py,,merge_aug_masks,#Any#Any#Any#Any#,81

Before Change

    Returns:
        tuple: (bboxes, scores)
    
    recovered_masks = [
        mask if not img_info[0]["flip"] else mask[..., ::-1]
        for mask, img_info in zip(aug_masks, img_metas)
    ]
    if weights is None:
        merged_masks = np.mean(recovered_masks, axis=0)
    else:
        merged_masks = np.average(

After Change

        flip = img_info[0]["flip"]
        flip_direction = img_info[0]["flip_direction"]
        if flip:
            if flip_direction == "horizontal":
                mask = mask[:, :, :, ::-1]
            elif flip_direction == "vertical":
                mask = mask[:, :, ::-1, :]
            else:
                raise ValueError(
                    f"Invalid flipping direction "{flip_direction}"")
        recovered_masks.append(mask)

    if weights is None:
        merged_masks = np.mean(recovered_masks, axis=0)

Instances