Commit a8cdb970 authored by dcuai's avatar dcuai
Browse files

Merge branch 'mmpretrain' into 'master'

updated readme for full imagenet dataset

See merge request !4
parents 205ed3b1 ecc9a21d
...@@ -85,13 +85,49 @@ pip install -r requirements.txt ...@@ -85,13 +85,49 @@ pip install -r requirements.txt
## 数据集 ## 数据集
### ImageNet
在本项目中可以使用ImageNet数据集。下载ImageNet数据集:https://image-net.org/ 在本项目中可以使用ImageNet数据集。下载ImageNet数据集:https://image-net.org/
SCNet快速下载链接[imagenet-2012](http://113.200.138.88:18080/aidatasets/project-dependency/imagenet-2012) 可于SCNet快速下载[imagenet-2012](http://113.200.138.88:18080/aidatasets/project-dependency/imagenet-2012)
下载其中的ILSVRC2012_img_train.tar和ILSVRC2012_img_val.tar并按照以下方式解包
```bash
cd mmpretrain-mmcv/data/imagenet
mkdir train && cd train
tar -xvf ILSVRC2012_img_train.tar
```
解包后是1000个tar文件,每个tar对应了一个类别,解包至对应文件夹
```bash
for tarfile in *.tar; do
dirname="${tarfile%.tar}"
mkdir "$dirname"
tar -xvf "$tarfile" -C "$dirname"
done
```
目录结构如下
```
data
└── imagenet
├── train
│   ├── n01440764
│   │   ├── n01440764_10026.JPEG
│   │   ├── n01440764_10027.JPEG
├──val
│   ├── n01440764
│   │   ├── ILSVRC2012_val_00000293.JPEG
```
### tiny-imagenet-200 ### tiny-imagenet-200
由于imagenet完整数据集较大,可以使用[tiny-imagenet-200](http://cs231n.stanford.edu/tiny-imagenet-200.zip)进行测试,此时需要对配置脚本进行一些修改: 由于ImageNet完整数据集较大,可以使用[tiny-imagenet-200](http://cs231n.stanford.edu/tiny-imagenet-200.zip)进行测试,此时需要对配置脚本进行一些修改:
- dataset配置文件(configs/\_\_base\_\_/datasets/xxx.py)中,需要对以下字段进行修改 - dataset配置文件(configs/\_\_base\_\_/datasets/xxx.py)中,需要对以下字段进行修改
...@@ -172,7 +208,7 @@ bash tools/dist_train.sh configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py 8 ...@@ -172,7 +208,7 @@ bash tools/dist_train.sh configs/mobilenet_v2/mobilenet-v2_8xb32_in1k.py 8
### 热点行业 ### 热点行业
制造,能源,交通,网安 制造,能源,交通,网安,安防
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
...@@ -181,5 +217,3 @@ https://developer.hpccube.com/codes/modelzoo/mobilenetv2_mmcv ...@@ -181,5 +217,3 @@ https://developer.hpccube.com/codes/modelzoo/mobilenetv2_mmcv
## 参考资料 ## 参考资料
https://github.com/open-mmlab/mmpretrain https://github.com/open-mmlab/mmpretrain
mmpretrain-mmcv @ 78daaf2c
Subproject commit 0d5d0b15d9d734bd95b183584bfe311fb70ede72 Subproject commit 78daaf2cbdb094afe7a1be6ff30bad6cc455acd8
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment