提交 3d3d0d31 编写于 作者: 文幕地方's avatar 文幕地方

fix vis error

上级 13c8569d
...@@ -148,7 +148,7 @@ def main(args): ...@@ -148,7 +148,7 @@ def main(args):
bbox_list_str)) bbox_list_str))
if len(bbox_list) > 0 and len(bbox_list[0]) == 4: if len(bbox_list) > 0 and len(bbox_list[0]) == 4:
img = draw_rectangle(image_file, pred_res['cell_bbox']) img = draw_rectangle(image_file, bbox_list)
else: else:
img = utility.draw_boxes(img, bbox_list) img = utility.draw_boxes(img, bbox_list)
img_save_path = os.path.join(args.output, img_save_path = os.path.join(args.output,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册