提交 1c43e8bb 编写于 作者: W WenmuZhou

Adapt to windows

上级 a44c2a69
...@@ -132,7 +132,7 @@ def main(args): ...@@ -132,7 +132,7 @@ def main(args):
valid_image_file_list.append(image_file) valid_image_file_list.append(image_file)
img_list.append(img) img_list.append(img)
try: try:
img_list, cls_res, elapse = text_classifier(img_list) img_list, cls_res, predict_time = text_classifier(img_list)
except Exception as e: except Exception as e:
print(e) print(e)
logger.info( logger.info(
...@@ -143,10 +143,10 @@ def main(args): ...@@ -143,10 +143,10 @@ def main(args):
"Please set --rec_image_shape='3,32,100' and --rec_char_type='en' ") "Please set --rec_image_shape='3,32,100' and --rec_char_type='en' ")
exit() exit()
for ino in range(len(img_list)): for ino in range(len(img_list)):
print("Predicts of %s:%s" % (valid_image_file_list[ino], cls_res[ino])) print("Predicts of {}:{}".format(valid_image_file_list[ino], cls_res[
print("Total predict time for %d images, cost: %.3f" % ino]))
(len(img_list), elapse)) print("Total predict time for {} images, cost: {:.3f}".format(
len(img_list), predict_time))
if __name__ == "__main__":
if __name__ == "__main__":
main(utility.parse_args()) main(utility.parse_args())
...@@ -178,10 +178,11 @@ if __name__ == "__main__": ...@@ -178,10 +178,11 @@ if __name__ == "__main__":
if count > 0: if count > 0:
total_time += elapse total_time += elapse
count += 1 count += 1
print("Predict time of %s:" % image_file, elapse) print("Predict time of {}: {}".format(image_file, elapse))
src_im = utility.draw_text_det_res(dt_boxes, image_file) src_im = utility.draw_text_det_res(dt_boxes, image_file)
img_name_pure = os.path.split(image_file)[-1] img_name_pure = os.path.split(image_file)[-1]
img_path = os.path.join(draw_img_save, "det_res_%s" % img_name_pure) img_path = os.path.join(draw_img_save,
"det_res_{}".format(img_name_pure))
cv2.imwrite(img_path, src_im) cv2.imwrite(img_path, src_im)
if count > 1: if count > 1:
print("Avg Time:", total_time / (count - 1)) print("Avg Time:", total_time / (count - 1))
...@@ -145,9 +145,10 @@ def main(args): ...@@ -145,9 +145,10 @@ def main(args):
"Please set --rec_image_shape='3,32,100' and --rec_char_type='en' ") "Please set --rec_image_shape='3,32,100' and --rec_char_type='en' ")
exit() exit()
for ino in range(len(img_list)): for ino in range(len(img_list)):
print("Predicts of %s:%s" % (valid_image_file_list[ino], rec_res[ino])) print("Predicts of {}:{}".format(valid_image_file_list[ino], rec_res[
print("Total predict time for %d images, cost: %.3f" % ino]))
(len(img_list), predict_time)) print("Total predict time for {} images, cost: {:.3f}".format(
len(img_list), predict_time))
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册