Skip to content

Commit c55d7a7

Browse files
authored
[Feature] support mim download dataset (#3089)
## 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
1 parent b633093 commit c55d7a7

File tree

3 files changed

+81
-1
lines changed

3 files changed

+81
-1
lines changed

MANIFEST.in

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
include requirements/*.txt
22
include mmseg/.mim/model-index.yml
3+
include mmaction/.mim/dataset-index.yml
34
recursive-include mmseg/.mim/configs *.py *.yaml
45
recursive-include mmseg/.mim/tools *.py *.sh

dataset-index.yml

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,79 @@
1+
ade20k:
2+
dataset: ADE20K_2016
3+
download_root: data
4+
data_root: data/ade
5+
6+
cityscapes:
7+
dataset: CityScapes
8+
download_root: data
9+
data_root: data/cityscapes
10+
11+
voc2012:
12+
dataset: PASCAL_VOC2012
13+
download_root: data
14+
data_root: data/VOCdevkit/VOC2012
15+
16+
cocostuff:
17+
dataset: COCO-Stuff
18+
download_root: data
19+
data_root: data/coco_stuff164k
20+
21+
mapillary:
22+
dataset: Mapillary
23+
download_root: data
24+
data_root: data/mapillary
25+
26+
pascal_context:
27+
dataset: VOC2010
28+
download_root: data
29+
data_root: data/VOCdevkit/VOC2010
30+
31+
isaid:
32+
dataset: iSAID
33+
download_root: data
34+
data_root: data/iSAID
35+
36+
isprs_potsdam:
37+
dataset: ISPRS_Potsdam
38+
download_root: data
39+
data_root: data/potsdam
40+
41+
loveda:
42+
dataset: LoveDA
43+
download_root: data
44+
data_root: data/loveDA
45+
46+
chase_db1:
47+
dataset: CHASE_DB1
48+
download_root: data
49+
data_root: data/CHASE_DB1
50+
51+
drive:
52+
dataset: DRIVE
53+
download_root: data
54+
data_root: data/DRIVE
55+
56+
hrf:
57+
dataset: HRF
58+
download_root: data
59+
data_root: data/HRF
60+
61+
stare:
62+
dataset: STARE
63+
download_root: data
64+
data_root: data/STARE
65+
66+
synapse:
67+
dataset: SurgVisDom
68+
download_root: data
69+
data_root: data/synapse
70+
71+
refuge:
72+
dataset: REFUGE_Challenge
73+
download_root: data
74+
data_root: data/REFUGE
75+
76+
lip:
77+
dataset: LIP
78+
download_root: data
79+
data_root: data/LIP

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ def add_mim_extension():
123123
else:
124124
return
125125

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

0 commit comments

Comments
 (0)