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)
