Skip to content

Support encoded RLE format in for COCO segmentations#8387

Merged
NicolasHug merged 3 commits intopytorch:mainfrom
NicolasHug:aelfjnalefjnaef
Jun 4, 2024
Merged

Support encoded RLE format in for COCO segmentations#8387
NicolasHug merged 3 commits intopytorch:mainfrom
NicolasHug:aelfjnalefjnaef

Conversation

@NicolasHug
Copy link
Copy Markdown
Member

@NicolasHug NicolasHug commented Apr 19, 2024

Closes #8351

This lets our COCO wrapper support segmentations that are already in "encoded RLE" format.

@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Apr 19, 2024

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/8387

Note: Links to docs will display an error until the docs builds have been completed.

❌ 11 New Failures, 8 Unrelated Failures

As of commit 2f7c1a5 with merge base 26af015 (image):

NEW FAILURES - The following jobs have failed:

FLAKY - The following jobs failed but were likely due to flakiness present on trunk:

BROKEN TRUNK - The following jobs failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

This comment was automatically generated by Dr. CI and updates every 15 minutes.

Comment thread test/test_datasets.py
with pytest.raises(ValueError, match="Index must be of type integer"):
dataset[:2]

def test_segmentation_kind(self):
Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Really not a fan of that test's logic, but I don't know how to better test that. @pmeier LMK if there are better options?

@NicolasHug NicolasHug merged commit f7d9e75 into pytorch:main Jun 4, 2024
@NicolasHug NicolasHug deleted the aelfjnalefjnaef branch June 4, 2024 12:01
facebook-github-bot pushed a commit that referenced this pull request Jun 7, 2024
Reviewed By: vmoens

Differential Revision: D58283856

fbshipit-source-id: 98805162e3209173811108468037ea490d123082
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

support compact/encoded RLE masks in COCO dataset wrapper

2 participants