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

Before Change

        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")


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

After Change

                          draw_lines=draw_lines, line_thickness=line_thickness,
                          draw_endpoints=draw_endpoints, diameter_endpoint=circle_thickness)

    classes = [(0, 0, 0)]
    if draw_baselines:
        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