Skip to content

coco mask converting #1195

@IgorDavidyuk

Description

@IgorDavidyuk

masks = masks.permute(0, 2, 1).contiguous().permute(0, 2, 1)

Docs for MASK R-CNN claims that target should contain 'masks (UInt8Tensor[N, 1, H, W])'
But the line of code above treats it like tensor with 3 dimensions. Catching errors.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions