[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

- [ ] https://github.com/open-mmlab/mim/pull/212
pull/3151/head
谢昕辰 2023-06-28 17:57:22 +08:00 committed by GitHub
parent b633093c27
commit c55d7a721e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 81 additions and 1 deletions

View File

@ -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
dataset-index.yml 100644
View File

@ -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

View File

@ -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)