Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Custom dataset with gt_masks #4927

Open
sedol1339 opened this issue Apr 26, 2023 · 0 comments
Open

Custom dataset with gt_masks #4927

sedol1339 opened this issue Apr 26, 2023 · 0 comments
Labels
documentation Problems about existing documentation or comments

Comments

@sedol1339
Copy link

sedol1339 commented Apr 26, 2023

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?

@sedol1339 sedol1339 added the documentation Problems about existing documentation or comments label Apr 26, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
documentation Problems about existing documentation or comments
Projects
None yet
Development

No branches or pull requests

1 participant