facebookresearch / facebookresearch/detectron2
Custom dataset with gt_masks
- 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
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