RuntimeError in: mmdet/core/anchor/anchor_target.py

[Randomless]

RuntimeError: Expected object of scalar type Byte but got scalar type Bool for argument #2

in: mmdet/core/anchor/anchor_target.py

def anchor_inside_flags(flat_anchors, valid_flags, img_shape,
                        allowed_border=0):
    img_h, img_w = img_shape[:2]
    allowed_border=-1
    if allowed_border >= 0:
        inside_flags = valid_flags & \
            (flat_anchors[:, 0] >= -allowed_border) & \
            (flat_anchors[:, 1] >= -allowed_border) & \
            (flat_anchors[:, 2] < img_w + allowed_border) & \
            (flat_anchors[:, 3] < img_h + allowed_border)
    else:
        inside_flags = valid_flags
    return inside_flags

[csuhan]

open-mmlab/mmdetection#3587

[zf020114]

if allowed_border >= 0:
    inside_flags = valid_flags & \
                   (flat_anchors[:, 0] >= -allowed_border).type(torch.uint8) & \
                   (flat_anchors[:, 1] >= -allowed_border).type(torch.uint8) & \
                   (flat_anchors[:, 2] < img_w + allowed_border).type(torch.uint8) & \
                   (flat_anchors[:, 3] < img_h + allowed_border).type(torch.uint8)