From f44a718ce962f9be5397af96d342321ef0f058d1 Mon Sep 17 00:00:00 2001
From: hysunflower <nothing_upc@163.com>
Date: Wed, 24 Nov 2021 02:43:45 +0000
Subject: [PATCH] update_scripts_benchmark

---
 benchmark/prepare_data.sh | 11 +++++++++++
 1 file changed, 11 insertions(+)

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}