PaddleClas/benchmark/prepare_data.sh

23 lines
591 B
Bash
Raw Normal View History

2021-09-17 11:14:28 +08:00
#!/bin/bash
2021-09-30 11:39:54 +08:00
dataset_url=$1
2021-09-17 11:14:28 +08:00
2021-11-24 10:43:45 +08:00
package_check_list=(imageio tqdm Cython pycocotools tb_paddle scipy pandas wget h5py sklearn opencv-python visualdl)
for package in ${package_check_list[@]}; do
if python -c "import ${package}" >/dev/null 2>&1; then
echo "${package} have already installed"
else
echo "${package} NOT FOUND"
pip install ${package}
echo "${package} installed"
fi
done
2021-09-17 11:14:28 +08:00
cd dataset
rm -rf ILSVRC2012
2021-09-30 11:39:54 +08:00
wget -nc ${dataset_url}
tar xf ILSVRC2012_val.tar
ln -s ILSVRC2012_val ILSVRC2012
cd ILSVRC2012
ln -s val_list.txt train_list.txt
2021-09-17 11:14:28 +08:00
cd ../../