Skip to content

Train MASK-RCNN using RLE-format maskΒ #2758

@KevinZhoutianyi

Description

@KevinZhoutianyi

πŸ“š Documentation

  1. BACKGROUND: I have more than one hundred objects on one image. And there is only one class. For each image, I got one ground truth mask for all objects
  2. I follow the "train on custom dataset" part in the tutorial (https://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5) and I want to use bitmask instead of polygons.
  3. I use pycocotools.encode to compress my ground truth mask to RLE format. And I assign that RLE-format mask to each "segmentation" in "annotations"
  4. Finally, it is really slow when training. I guess it is because the mask is large, and there is a same mask for each object.

Is there any mistake in my steps? Or how should I use my ground truth mask image?
Thanks for your help

Metadata

Metadata

Assignees

No one assigned

    Labels

    documentationProblems about existing documentation or comments

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions