facebookresearch / facebookresearch/detectron2

Custom dataset with gt_masks

Open
#4,927 0 comments 0 reactions 0 assignees View on GitHub
documentation
Dominant language
Python
Stars
34.7k
Forks
7.9k
PR merge metrics
No merged PRs in 30d

Description

Hello, I want to fine-tune mask R-CNN on a custom dataset with masks. I have these masks on a disk. So, I guess I need to add a field gt_masks in my dataloader.

However, I did not find documentation about the format of gt_masks: should it be if CHW, HWC or HW format, in float 0-1 or uint8 0-255, etc?

I guess that the only way to get this information is to create a dataloader for some of built-in datasets. However, all these datasets are large. For example, creating a dataloader for "coco_2014_minival_100" requires images in "datasets/coco/val2014" folder, so I need to wait half a day to download 6GB val2014.zip and unpack.

Maybe i'm doing something wrong? How to solve my problem in a normal way?

Contributor guide

Open the contributing guide

Research direction

Start by inspecting the custom dataloader guidance and the built-in dataset examples mentioned in the issue, especially how gt_masks is represented for Mask R-CNN. Document the expected mask shape and value type, and explain a normal way to inspect or validate the format without downloading the full COCO dataset.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
documentation, machine-learning
Issue type
Documentation
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.