-
Notifications
You must be signed in to change notification settings - Fork 411
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
Add SODA-A dataset #2575
Add SODA-A dataset #2575
Conversation
@shaunyuan22 Thank you for this nice dataset and all the work. We aim to make the dataset more easily usable in torchgeo, and would appreciate it if you have any comments, corrections etc. |
Open question how to deal with the polygons into a common oriented bounding box schema. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Asked about oriented bboxes on Slack. We may need to add support to Kornia for this ourselves. Until Kornia supports it natively, I guess it doesn't matter what the format looks like. But let's use the same key names that Kornia uses.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Most of my remaining comments are optional pedantic things, this is 99% ready to merge.
array: np.typing.NDArray[np.int_] = np.array(img.convert('RGB')) | ||
tensor: Tensor = torch.from_numpy(array) | ||
# Convert from HxWxC to CxHxW | ||
tensor = tensor.permute((2, 0, 1)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Might be more clear if you use einops.rearrange here
This PR adds the SODA-A dataset. Dataset rehosted on HF.
Dataset features:
Dataset format:
TODOS:
Annotations/train/01874.json
:Example plot:
