提交 ffca17ac 编写于 作者: G gaotingquan 提交者: Tingquan Gao

replace the dataset with full ILSVRC2012

上级 e355f2e7
...@@ -324,7 +324,7 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then ...@@ -324,7 +324,7 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then
python_name=$(func_parser_value "${lines[2]}") python_name=$(func_parser_value "${lines[2]}")
inference_model_url=$(func_parser_value "${lines[10]}") inference_model_url=$(func_parser_value "${lines[10]}")
tar_name=$(func_get_url_file_name "$inference_model_url") tar_name=$(func_get_url_file_name "$inference_model_url")
${python_name} -m pip install onnx ${python_name} -m pip install onnx
${python_name} -m pip install paddle2onnx ${python_name} -m pip install paddle2onnx
${python_name} -m pip install onnxruntime ${python_name} -m pip install onnxruntime
...@@ -368,13 +368,10 @@ if [[ ${MODE} = "benchmark_train" ]]; then ...@@ -368,13 +368,10 @@ if [[ ${MODE} = "benchmark_train" ]]; then
ln -s demo_test.txt val_list.txt ln -s demo_test.txt val_list.txt
else else
rm -rf ILSVRC2012 rm -rf ILSVRC2012
wget -nc https://paddle-imagenet-models-name.bj.bcebos.com/data/ImageNet1k/ILSVRC2012_val.tar wget -nc https://paddle-wheel.bj.bcebos.com/benchmark/ILSVRC2012_w.tar
tar xf ILSVRC2012_val.tar tar xf ILSVRC2012_w.tar
ln -s ILSVRC2012_val ILSVRC2012 ln -s ILSVRC2012_w ILSVRC2012
cd ILSVRC2012
rm -rf train_list.txt
ln -s val_list.txt train_list.txt
fi fi
cd ../../ cd ../
fi fi
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册