def generate_samples(filename, output='./output.mp4'):
if not clb.initialized():
clb.find_pictures(directory='./camera_cal/')
clb.calibrate_camera(9, 6)
m = model.CarModel()
m.load()
t = track.FrameVehiclePipeline(m, shape=(720, 1280))
video = VideoFileClip(filename)
out = video.fl_image(t.samples)
out.write_videofile(output, audio=False)
def single_image(filename, model_path='./data/model.p'):
if not clb.initialized():
clb.find_pictures(directory='./camera_cal/')
clb.calibrate_camera(9, 6)
im = common.load_image(filename, color='RGB')
common.show_image(im)
m = model.CarModel()
m.load(filename=model_path)
t = track.FrameVehiclePipeline(m, shape=im.shape[:2])
t.process(im, show=True)
