Add Python Inference Models

master
tunm 2 years ago
parent 55626034a3
commit 58ad4ef736

@ -21,7 +21,7 @@ class LPRMultiTaskPipeline(object):
score = out[4] score = out[4]
land_marks = out[5:13].reshape(4, 2).astype(int) land_marks = out[5:13].reshape(4, 2).astype(int)
layer_num = int(out[13]) layer_num = int(out[13])
print(layer_num) # print(layer_num)
pad = get_rotate_crop_image(image, land_marks) pad = get_rotate_crop_image(image, land_marks)
if layer_num == DOUBLE: if layer_num == DOUBLE:
# double # double

Loading…
Cancel
Save