forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix tools/infer/utility.py

This commit is contained in:
LDOUBLEV 2020-05-11 15:33:54 +08:00
parent 3f2d384faa
commit af39902217
1 changed files with 2 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tools/infer/utility.py
View File

@ -99,7 +99,8 @@ def create_predictor(args, mode):
config.disable_gpu()
config.disable_glog_info()
config.switch_ir_optim(args.ir_optim)
# config.switch_ir_optim(args.ir_optim)
# if args.use_tensorrt:
# config.enable_tensorrt_engine(
# precision_mode=AnalysisConfig.Precision.Half
@ -127,25 +128,3 @@ def draw_text_det_res(dt_boxes, img_path):
cv2.polylines(src_im, [box], True, color=(255, 255, 0), thickness=2)
img_name_pure = img_path.split("/")[-1]
cv2.imwrite("./output/%s" % img_name_pure, src_im)
if __name__ == '__main__':
args = parse_args()
args.use_gpu = False
root_path = "/Users/liuweiwei06/Desktop/TEST_CODES/icode/baidu/personal-code/PaddleOCR/"
args.det_model_dir = root_path + "test_models/public_v1/ch_det_mv3_db"
predictor, input_tensor, output_tensors = create_predictor(args, mode='det')
print("det input", predictor.get_input_names())
print("det output", predictor.get_output_names())
# print(predictor.program(), file=open("det_program.txt", 'w'))
outputs = []
for output_tensor in output_tensors:
output = output_tensor.copy_to_cpu()
outputs.append(output)
args.rec_model_dir = root_path + "test_models/public_v1/ch_rec_mv3_crnn/"
rec_predictor, input_tensor, output_tensors = create_predictor(
args, mode='rec')
print("rec input", rec_predictor.get_input_names())
print("rec output", rec_predictor.get_output_names())