Add the ADE20K dataset in TorchVision
Open
Nobody has claimed this yet.
module: datasets
prototype
- Dominant language
- Python
- Stars
- 17.9k
- Forks
- 7.3k
- Avg merge
- 1d 15h
- Merged PRs (30d)
- 13
Description
🚀 The feature
ADE20K is a dataset introduced in 2017 for Object Detection and Segmentation (code here). It has 371 citations and it's become more and more popular as a benchmark.
We should consider adding it to TorchVision using the new Dataset API.
cc @pmeier @bjuncek
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reviewing TorchVision's existing dataset implementations and the new Dataset API. Determine how ADE20K's object-detection and segmentation data should be represented, then verify that the dataset can be downloaded, loaded, and exercised through the relevant dataset tests. The issue does not name specific files or tests.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- computer-vision, machine-learning
- Issue type
- Feature
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 28/100