liang_chaoming@huawei.com
48 lines
1.0 KiB
Bash
48 lines
1.0 KiB
Bash
#!/bin/bash
|
|
|
|
# Get COCO 2014 data sets
|
|
|
|
if [ $1 == 'YoLov3' ];then
|
|
echo 111
|
|
mkdir -p /home/datasets/coco
|
|
pushd /home/datasets/coco
|
|
|
|
curl -O http://images.cocodataset.org/zips/train2014.zip
|
|
unzip train2014.zip
|
|
|
|
curl -O http://images.cocodataset.org/zips/val2014.zip
|
|
unzip val2014.zip
|
|
|
|
curl -O http://images.cocodataset.org/annotations/annotations_trainval2014.zip
|
|
unzip annotations_trainval2014.zip
|
|
|
|
# Get bert/cule data sets
|
|
elif [ $1 == 'Bert' ];then
|
|
echo 222
|
|
mkdir -p /home/datasets/Bertdata
|
|
pushd /home/datasets/Bertdata
|
|
|
|
curl -O xxxxxxxxxxx
|
|
tar xxxxx
|
|
|
|
# Get imagenet_TF data sets
|
|
else
|
|
echo 333
|
|
mkdir -p /home/datasets/imagenet_TF
|
|
pushd /home/datasets/imagenet_TF
|
|
|
|
|
|
curl -O http://www.image-net.org/challenges/LSVRC/2012/nnoupb/ILSVRC2012_img_val.tar
|
|
tar xvf ILSVRC2012_img_val.tar
|
|
|
|
curl -O http://www.image-net.org/challenges/LSVRC/2012/nnoupb/ILSVRC2012_img_train.tar
|
|
tar xvf ILSVRC2012_img_train.tar
|
|
|
|
curl -O http://www.image-net.org/challenges/LSVRC/2012/nnoupb/ILSVRC2012_bbox_train_v2.tar
|
|
tar xvf ILSVRC2012_bbox_train_v2.tar
|
|
fi
|
|
|
|
|
|
|
|
popd
|