82d2e9ea62f1d776331e843266a31fe076d249b1,src/data/generation/generate_crosswalk_images.py,,run,#Any#,7

Before Change

def run(args):
    bbox = Bbox.from_lbrt(args.bbox[0], args.bbox[1], args.bbox[2], args.bbox[3])
    crosswalkCollector = CrosswalkCollector(bbox=bbox, hdf5_file=args.convnet, image_dir=args.image_dir)
    crosswalkCollector.run()


def mainfunc():
    parser = argparse.ArgumentParser(description="Generate crosswalk images.", )

After Change

    bboxes = []
    if args.csv is not None:
        if os.path.exists(args.csv):
            with open(args.csv, "rb") as csvfile:
                reader = csv.reader(csvfile, delimiter=",")
                next(reader)
                for row in reader:
                    bbox = Bbox.from_lbrt(float(row[1]), float(row[2]), float(row[3]), float(row[4]))
                    bboxes.append(bbox)
        else:
            raise Exception("CSV file " + args.csv + " does not exist!")
    else:
        bboxes.append(Bbox.from_lbrt(args.bbox[0], args.bbox[1], args.bbox[2], args.bbox[3]))

Instances