diff --git a/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_full_1p.sh b/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_full_1p.sh index f1c73727ea8d482d29ca8b9d54e9de223ec8cc5a..03bdd98e7f376c74fb52ac606dcbbab771098e47 100644 --- a/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_full_1p.sh +++ b/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_full_1p.sh @@ -111,15 +111,25 @@ start_time=$(date +%s) train_epochs=30 batch_size=4 + if [ x"${modelarts_flag}" != x ]; then - python3.7 ./train.py --data_path=${data_path}/dataset/MPII --output_path=${output_path} \ + python3.7 ./train.py --data_path=${data_path}/dataset/MPII \ + --label_path=${data_path}/dataset/MPII/mpii_human_pose_v1_u12_2/mpii_human_pose_v1_u12_1.mat \ + --prob_model_path=${data_path}/data/prob_model/prob_model_params.mat \ + --init_session_path=${data_path}/data/init_session/init \ + --output_path=${output_path} \ --epochs=${train_epochs} --batch_size=${batch_size} else - python3.7 ./train.py --data_path=${data_path}/dataset/MPII --output_path=${output_path} \ + python3.7 ./train.py --data_path=${data_path}/dataset/MPII \ + --label_path=${data_path}/dataset/MPII/mpii_human_pose_v1_u12_2/mpii_human_pose_v1_u12_1.mat \ + --prob_model_path=${data_path}/data/prob_model/prob_model_params.mat \ + --init_session_path=${data_path}/data/init_session/init \ + --output_path=${output_path} \ --epochs=${train_epochs} --batch_size=${batch_size} 1>${print_log} 2>&1 fi + # 性能相关数据计算 #读取iteration/s数据(ITPS),再计算StepTime = 1/ITPS; FPS=BATCH_SIZE * ITPS diff --git a/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_performance_1p.sh b/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_performance_1p.sh index d8c9df87803ca8bf2814497d9e4bcf92b5e3b79a..35cc4e20ead0530daa5a6f2be3a459ff17a735cb 100644 --- a/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_performance_1p.sh +++ b/TensorFlow/contrib/cv/LiftingFromTheDeep_ID0891_for_Tensorflow/test/train_performance_1p.sh @@ -111,15 +111,25 @@ start_time=$(date +%s) train_epochs=1 batch_size=4 + if [ x"${modelarts_flag}" != x ]; then - python3.7 ./train.py --data_path=${data_path}/dataset/MPII--output_path=${output_path} \ + python3.7 ./train.py --data_path=${data_path}/dataset/MPII \ + --label_path=${data_path}/dataset/MPII/mpii_human_pose_v1_u12_2/mpii_human_pose_v1_u12_1.mat \ + --prob_model_path=${data_path}/data/prob_model/prob_model_params.mat \ + --init_session_path=${data_path}/data/init_session/init \ + --output_path=${output_path} \ --epochs=${train_epochs} --batch_size=${batch_size} else - python3.7 ./train.py --data_path=${data_path}/dataset/MPII --output_path=${output_path} \ + python3.7 ./train.py --data_path=${data_path}/dataset/MPII \ + --label_path=${data_path}/dataset/MPII/mpii_human_pose_v1_u12_2/mpii_human_pose_v1_u12_1.mat \ + --prob_model_path=${data_path}/data/prob_model/prob_model_params.mat \ + --init_session_path=${data_path}/data/init_session/init \ + --output_path=${output_path} \ --epochs=${train_epochs} --batch_size=${batch_size} 1>${print_log} 2>&1 fi + # 性能相关数据计算 #读取iteration/s数据(ITPS),再计算StepTime = 1/ITPS; FPS=BATCH_SIZE * ITPS @@ -129,7 +139,7 @@ FPS=`awk 'BEGIN{printf "%.2f\n", '${batch_size}'*'${ITPS}'}'` # 精度相关数据计算 -train_accuracy=`grep "Validation PCKh@0.5:" ${print_log} | tail -n 1| awk '{print $4}' | cut -c 10- | awk '{sum+=$1} END {print sum/NR}'` +#train_accuracy=`grep "Validation PCKh@0.5:" ${print_log} | tail -n 1| awk '{print $4}' | cut -c 10- | awk '{sum+=$1} END {print sum/NR}'` # 提取所有loss打印信息 grep "loss=" ${print_log} | awk -F "=" '{print $2}' > ./test/output/${ASCEND_DEVICE_ID}/my_output_loss.txt @@ -174,7 +184,7 @@ echo "Final Performance sec/step : $StepTime" echo "E2E Training Duration sec : $e2e_time" # 输出训练精度 -echo "Final Train Accuracy : ${train_accuracy}" +#echo "Final Train Accuracy : ${train_accuracy}" # 最后一个迭代loss值,不需要修改 ActualLoss=(`awk 'END {print $NF}' $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}_loss.txt`) @@ -188,4 +198,4 @@ echo "CaseName = ${CaseName}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}. echo "ActualFPS = ${FPS}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log echo "TrainingTime = ${StepTime}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log echo "ActualLoss = ${ActualLoss}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log -#echo "E2ETrainingTime = ${e2e_time}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log +echo "E2ETrainingTime = ${e2e_time}" >> $cur_path/output/$ASCEND_DEVICE_ID/${CaseName}.log