diff --git a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/README.md b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/README.md index 4b581197385b3ff6f90f2d11107275d2ccf33b53..e7d80b211622743e1f5f787e761b90255adec63e 100644 --- a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/README.md +++ b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/README.md @@ -50,7 +50,28 @@ python3 pb_to_pbtxt.py 20180408-102900.pb 1.删除第一个节点batch_size -2.第二个节点phase_train的op修改为const +2.第二个节点phase_train修改为: +node { + name: "phase_train" + op: "Const" + attr { + key: "dtype" + value { + type: DT_BOOL + } + } + attr { + key: "value" + value { + tensor { + dtype: DT_BOOL + tensor_shape { + } + bool_val: false + } + } + } +} 3.删除第三个节点batch_join/fifo_queue diff --git a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data.py b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data.py index 5d839bb5a818290bab2c40deaca94c0a965ec0ad..0eedf0c7eed64693aee1ddcc94192fabcad8d6c7 100644 --- a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data.py +++ b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data.py @@ -149,7 +149,7 @@ def main(args): control_placeholder = tf.compat.v1.placeholder(tf.int32, shape=(None, 1), name='control') phase_train_placeholder = tf.compat.v1.placeholder(tf.bool, name='phase_train') - nrof_preprocess_threads = 4 + nrof_preprocess_threads = 1 image_size = (args.image_size, args.image_size) eval_input_queue = data_flow_ops.FIFOQueue(capacity=2000000, dtypes=[tf.string, tf.int32, tf.int32], diff --git a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_forquant.py b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_fp32.py similarity index 97% rename from ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_forquant.py rename to ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_fp32.py index 8563095c50cc224d5390af262ca4e61253925334..f2e7b62b7f79a56dbc95430a7004e9ac9751761b 100644 --- a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_forquant.py +++ b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/preprocess_data_fp32.py @@ -150,7 +150,7 @@ def main(args): control_placeholder = tf.compat.v1.placeholder(tf.int32, shape=(None, 1), name='control') phase_train_placeholder = tf.compat.v1.placeholder(tf.bool, name='phase_train') - nrof_preprocess_threads = 4 + nrof_preprocess_threads = 1 image_size = (args.image_size, args.image_size) eval_input_queue = data_flow_ops.FIFOQueue(capacity=2000000, dtypes=[tf.string, tf.int32, tf.int32],