search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.18k stars 6.95k forks source link

Add the ADE20K dataset in TorchVision #5200

Open datumbox opened 2 years ago

datumbox commented 2 years ago

🚀 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

pmeier commented 2 years ago

This might be a good onboarding task for the new datasets.

NicolasHug commented 2 years ago

I'll do it

23pointsNorth commented 2 years ago

Is there a need for any support on this?

pmeier commented 2 years ago

@23pointsNorth We are currently halting further contributions to the prototype datasets, since there are some details that need to be figured out. If that is resolved, I'm guessing @NicolasHug can assign it to you given that he already did quite a few ports and no longer needs to be on-boarded.

kswain55 commented 2 years ago

Hi @datumbox, @pmeier! Are you planning to resume the process of adding ADE20K?