From 685fd05886a0b018cbf0ac9eb332bd59a5735589 Mon Sep 17 00:00:00 2001 From: alex Date: Tue, 15 Aug 2023 10:47:54 +0000 Subject: [PATCH] update ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/post2.py. Signed-off-by: alex --- .../built-in/cv/Facenet_for_ACL/script/post2.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/post2.py b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/post2.py index 4da9d73b0..8ae8c6b2a 100644 --- a/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/post2.py +++ b/ACL_TensorFlow/built-in/cv/Facenet_for_ACL/script/post2.py @@ -86,14 +86,14 @@ def main(args): out_image_name = os.path.join(args.input_dir,"{}_0.bin".format(str(i).zfill(6))) #print(out_image_name) emb = np.fromfile(out_image_name, dtype="float32").reshape(1, 512) - out_label_name = os.path.join(args.label_dir,"{}.bin".format(str(i).zfill(6))) - lab = np.fromfile(out_label_name, dtype="int32") + elif args.file_format == 'npy': out_image_name = os.path.join(args.input_dir, "{}_0.npy".format(str(i).zfill(6))) # print(out_image_name) emb = np.load(out_image_name) - out_label_name = os.path.join(args.label_dir, "{}.npy".format(str(i).zfill(6))) - lab = np.load(out_label_name) + + out_label_name = os.path.join(args.label_dir,"{}.bin".format(str(i).zfill(6))) + lab = np.fromfile(out_label_name, dtype="int32") ########## lab_array[lab] = lab -- Gitee