diff --git a/benchmark/prepare_data.sh b/benchmark/prepare_data.sh index 83a685698..411459cf4 100644 --- a/benchmark/prepare_data.sh +++ b/benchmark/prepare_data.sh @@ -1,6 +1,17 @@ #!/bin/bash dataset_url=$1 +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 + cd dataset rm -rf ILSVRC2012 wget -nc ${dataset_url}