From c55d7a721e9b0ca628e20eb23578bb7728f80465 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= <xiexinch@outlook.com>
Date: Wed, 28 Jun 2023 17:57:22 +0800
Subject: [PATCH] [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
---
 MANIFEST.in       |  1 +
 dataset-index.yml | 79 +++++++++++++++++++++++++++++++++++++++++++++++
 setup.py          |  2 +-
 3 files changed, 81 insertions(+), 1 deletion(-)
 create mode 100644 dataset-index.yml

diff --git a/MANIFEST.in b/MANIFEST.in
index 7734e71d5f..bd8c7c06d2 100644
--- a/MANIFEST.in
+++ b/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
diff --git a/dataset-index.yml b/dataset-index.yml
new file mode 100644
index 0000000000..dc235099fd
--- /dev/null
+++ b/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
diff --git a/setup.py b/setup.py
index a117afae7b..7316ed1c84 100755
--- a/setup.py
+++ b/setup.py
@@ -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)