def add_mark(img_file, csv_file, hasHeader=False, shape=Shape.CROSS, mark_color=(0, 255, 127, 200), hasProb=False): add_ground_truth_mark_help(img_file, csv_file, hasHeader=hasHeader, shape=shape, mark_color=mark_color, hasProb=hasProb)
def add_groundtruth_mark(im_dir, ground_truth_dir, hasHeader=False, shape=Shape.CROSS, mark_color=(0, 255, 127, 200), hasProb=False): im_files = [str(f) for f in Path(im_dir).glob('*.png')] for im_file in im_files: im_file_basename = os.path.basename(im_file).split('.')[0] ground_truth_file_path = os.path.join(ground_truth_dir, *im_file_basename.split("-")) ground_truth_file_path = "{}.csv".format(ground_truth_file_path) if not os.path.exists(ground_truth_file_path): print("{} doestn't exist".format(ground_truth_file_path)) continue add_ground_truth_mark_help(im_file, ground_truth_file_path, hasHeader=hasHeader, shape=shape)