468b7ab7a48a6712991dcbb4ec399a4dc95bd091,manage.py,,,#,94

Before Change

    if args["train"]:
        //Read in pictures and velocities and create a predictor
        if "indir" in args: indir = args["indir"]
        else: indir = None

        train(indir)

After Change

if __name__ == "__main__":

    session = args["--session"] or None
    model = args["--model"] or None

    print("session name: %s,  model name: %s " %(session, model))


    if args["drive"]:

        GLB.camera = settings.camera()
        GLB.controller = settings.controller()
        GLB.vehicle = settings.vehicle()
        
        GLB.recorder = settings.recorder(session)
        //GLB.recorder.create(session)

        if model is not None:
            print("Loading predictor model: %s" %model)
            GLB.predictor = settings.predictor()
            GLB.predictor.load(model)
        else:
            print("No model given. Auto mode will not work.")
            GLB.predictor = settings.predictor()
        drive_loop()


    elif args["train"]:
        //Read in pictures and velocities and create a predictor
        recorder = settings.recorder(session)
        predictor = settings.predictor()
        predictor.create(model)
        train(recorder, predictor)

Instances