fix vis error
@@ -148,7 +148,7 @@ def main(args):
bbox_list_str))
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:
img = utility.draw_boxes(img, bbox_list)
img_save_path = os.path.join(args.output,
