p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!17509 deeptext acc bug fix on 310 

From: @zhouneng2
Reviewed-by: @oacjiewen,@c_34
Signed-off-by: @c_34
This commit is contained in:
mindspore-ci-bot 2021-06-02 16:36:56 +08:00 committed by Gitee
parent 9cb53c0a5e ca7a79d17c
commit 8f89203a59
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/deeptext/postprocess.py
View File

@ -29,7 +29,7 @@ def get_pred(file, result_path):
all_bbox_file = os.path.join(result_path, file_name + "_0.bin")
all_label_file = os.path.join(result_path, file_name + "_1.bin")
all_mask_file = os.path.join(result_path, file_name + "_2.bin")
all_bbox = np.fromfile(all_bbox_file, dtype=np.float32).reshape(config.test_batch_size, 1000, 5)
all_bbox = np.fromfile(all_bbox_file, dtype=np.float16).reshape(config.test_batch_size, 1000, 5)
all_label = np.fromfile(all_label_file, dtype=np.int32).reshape(config.test_batch_size, 1000, 1)
all_mask = np.fromfile(all_mask_file, dtype=np.bool).reshape(config.test_batch_size, 1000, 1)