Skip to content

Commit

Permalink
[Feature] support mim download dataset (#3089)
Browse files Browse the repository at this point in the history
## Motivation

Please describe the motivation of this PR and the goal you want to
achieve through this PR.

## Modification

- add dataset-index.yml

## Dependencies

- [ ] open-mmlab/mim#212
  • Loading branch information
xiexinch committed Jun 28, 2023
1 parent b633093 commit c55d7a7
Show file tree
Hide file tree
Showing 3 changed files with 81 additions and 1 deletion.
1 change: 1 addition & 0 deletions MANIFEST.in
@@ -1,4 +1,5 @@
include requirements/*.txt
include mmseg/.mim/model-index.yml
include mmaction/.mim/dataset-index.yml
recursive-include mmseg/.mim/configs *.py *.yaml
recursive-include mmseg/.mim/tools *.py *.sh
79 changes: 79 additions & 0 deletions dataset-index.yml
@@ -0,0 +1,79 @@
ade20k:
dataset: ADE20K_2016
download_root: data
data_root: data/ade

cityscapes:
dataset: CityScapes
download_root: data
data_root: data/cityscapes

voc2012:
dataset: PASCAL_VOC2012
download_root: data
data_root: data/VOCdevkit/VOC2012

cocostuff:
dataset: COCO-Stuff
download_root: data
data_root: data/coco_stuff164k

mapillary:
dataset: Mapillary
download_root: data
data_root: data/mapillary

pascal_context:
dataset: VOC2010
download_root: data
data_root: data/VOCdevkit/VOC2010

isaid:
dataset: iSAID
download_root: data
data_root: data/iSAID

isprs_potsdam:
dataset: ISPRS_Potsdam
download_root: data
data_root: data/potsdam

loveda:
dataset: LoveDA
download_root: data
data_root: data/loveDA

chase_db1:
dataset: CHASE_DB1
download_root: data
data_root: data/CHASE_DB1

drive:
dataset: DRIVE
download_root: data
data_root: data/DRIVE

hrf:
dataset: HRF
download_root: data
data_root: data/HRF

stare:
dataset: STARE
download_root: data
data_root: data/STARE

synapse:
dataset: SurgVisDom
download_root: data
data_root: data/synapse

refuge:
dataset: REFUGE_Challenge
download_root: data
data_root: data/REFUGE

lip:
dataset: LIP
download_root: data
data_root: data/LIP
2 changes: 1 addition & 1 deletion setup.py
Expand Up @@ -123,7 +123,7 @@ def add_mim_extension():
else:
return

filenames = ['tools', 'configs', 'model-index.yml']
filenames = ['tools', 'configs', 'model-index.yml', 'dataset-index.yml']
repo_path = osp.dirname(__file__)
mim_path = osp.join(repo_path, 'mmseg', '.mim')
os.makedirs(mim_path, exist_ok=True)
Expand Down

0 comments on commit c55d7a7

Please sign in to comment.