diff --git a/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/set_conda.sh b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/set_conda.sh new file mode 100644 index 0000000000000000000000000000000000000000..55087d8622f46e055bb105a2acd1fe7006d4af07 --- /dev/null +++ b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/set_conda.sh @@ -0,0 +1,2 @@ +export PATH=/home/anaconda3/bin:$PATH +export LD_LIBRARY_PATH=/home/anaconda3/lib:$LD_LIBRARY_PATH \ No newline at end of file diff --git a/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/train_RT2_performance_1p.sh b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/train_RT2_performance_1p.sh index 5369e98f940e366793e194a802c22f88e1683971..620e8b48d03040b5438ecedf689a9db6b88e5fee 100644 --- a/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/train_RT2_performance_1p.sh +++ b/TensorFlow/contrib/cv/Pix2pose_ID1164_for_TensorFlow/test/train_RT2_performance_1p.sh @@ -81,7 +81,12 @@ do data_path=`echo ${para#*=}` elif [[ $para == --ckpt_path* ]];then ckpt_path=`echo ${para#*=}` - fi + elif [[ $para == --conda_name* ]];then + conda_name=`echo ${para#*=}` + source set_conda.sh + echo "conda_name: $conda_name" + source activate $conda_name + fi done # #校验是否传入data_path,不需要修改 # if [[$data_path == ""]];then