diff --git a/ACL_TensorFlow/built-in/cv/SSD-Resnet34_for_ACL/README.md b/ACL_TensorFlow/built-in/cv/SSD-Resnet34_for_ACL/README.md index 1be6517744c462a03fa43a7eccde61eb541d7d07..a3273483147aa565ee1c26c7d6303ae9a5a01db7 100644 --- a/ACL_TensorFlow/built-in/cv/SSD-Resnet34_for_ACL/README.md +++ b/ACL_TensorFlow/built-in/cv/SSD-Resnet34_for_ACL/README.md @@ -2,7 +2,7 @@ # SSD Resnet34 TensorFlow离线推理 -此链接提供Vgg16 TensorFlow模型在NPU上离线推理的脚本和方法 +此链接提供SSD Resnet34 TensorFlow模型在NPU上离线推理的脚本和方法 ## 注意 **此案例仅为您学习Ascend软件栈提供参考,不用于商业目的。** diff --git a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README.md b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README.md index 447b31905f330435d79b05360e8e458d3ebe1461..ee38418a8c162d07ed8fc0d52f008c809cd7267d 100644 --- a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README.md +++ b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README.md @@ -48,7 +48,8 @@ cd Modelzoo-TensorFlow/ACL/Official/cv/MTCNN_for_ACL ``` cd .. - python3 acltest.py ompath data_in_om data_out_om Ascend310P3 ./mtc_pnet.pb ./mtc_rnet.pb ./mtc_onet.pb + export DUMP_GE_GRAPH=2 + python3 acltest.py ompath data_in_om data_out_om Ascend310P3 ./mtcnn_pnet_tf.pb ./mtcnn_rnet_tf.pb ./mtcnn_onet_tf.pb ``` 注意: 默认情况下,“picture”目录中的图像已被推理。如果要替换图像,请替换“picture”目录中的图像。 diff --git a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README_EN.md b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README_EN.md index c715389be0c246449ed39eb61f00624a5f757eb8..df3f65321d8e0830f6b0f211ea24f79bd2b75162 100644 --- a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README_EN.md +++ b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/README_EN.md @@ -46,7 +46,8 @@ cd Modelzoo-TensorFlow/ACL/Official/cv/MTCNN_for_ACL ``` cd .. - python3 acltest.py ompath data_in_om data_out_om Ascend310P3 ./mtc_pnet.pb ./mtc_rnet.pb ./mtc_onet.pb + export DUMP_GE_GRAPH=2 + python3 acltest.py ompath data_in_om data_out_om Ascend310P3 ./mtcnn_pnet_tf.pb ./mtcnn_rnet_tf.pb ./mtcnn_onet_tf.pb ``` Notes: By default, image in the "picture" directory are inferred. If you want to replace image, replace the image in the "picture" directory. diff --git a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/acltest.py b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/acltest.py index 130be5ca18d97a50dbe06e29fe96fd66ae4a2003..4ff8838203e54580ad23a039a0d1c2e48e48e638 100644 --- a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/acltest.py +++ b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/acltest.py @@ -38,6 +38,7 @@ if input_mode=='1': os.system("python3 om_bm_predict.py {} {} {} {} {}".format(sys.argv[1],sys.argv[2],sys.argv[3],sys.argv[4],sys.argv[5])) os.system("python3 om_bm_predict_rnet.py {} {} {} {} {}".format(sys.argv[1],sys.argv[2],sys.argv[3],sys.argv[4],sys.argv[6])) boxes_c, landmarks = om_bm_predict_onet.do_bm_predict("{}".format(sys.argv[1]),"{}".format(sys.argv[2]),"{}".format(sys.argv[3]),"{}".format(sys.argv[4]),"{}".format(sys.argv[7])) + os.system("rm -rf ge_*txt") print(boxes_c) for i in range(boxes_c.shape[0]): bbox=boxes_c[i,:4] diff --git a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/output/.keep b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/output/.keep new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/picture/.keep b/ACL_TensorFlow/contrib/cv/MTCNN_for_ACL/picture/.keep new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391