3b6224bc88d9ae94837225d9aed8c41789c8ebda,exps/cbad/utils.py,,cbad_set_generator,#,101

Before Change

    if draw_endpoints:
        classes = np.stack([(0, 0, 0), DRAWING_COLOR_BASELINES, DRAWING_COLOR_POINTS])
        np.savetxt(os.path.join(output_dir, "classes.txt"), classes, fmt="%d")
    else:
        classes = np.stack([(0, 0, 0), DRAWING_COLOR_BASELINES])
        np.savetxt(os.path.join(output_dir, "classes.txt"), classes, fmt="%d")

After Change

        classes.append(DRAWING_COLOR_BASELINES)
    if draw_lines:
        classes.append(DRAWING_COLOR_LINES)
    if draw_endpoints:
        classes.append(DRAWING_COLOR_POINTS)

    np.savetxt(os.path.join(output_dir, "classes.txt"), classes, fmt="%d")


def draw_lines_fn(xml_filename: str, output_dir: str):

Instances