r = r - 30. * vis.lanes[2].shifted(-1).gaussian() r = r + 30. * feature.speed(0.5) r = r + 10. * vis.lanes[0].gaussian(10.) r = r + .1 * feature.control() vis.cars[1].reward = r vis.main_car = vis.cars[0] vis.paused = True vis.set_heat(r) #vis.set_heat(vis.lanes[0].gaussian()+vis.lanes[1].gaussian()+vis.lanes[2].gaussian()) #vis.set_heat(-vis.cars[1].traj.gaussian()+vis.lanes[0].gaussian()+vis.lanes[1].gaussian()+vis.lanes[2].gaussian()) vis.run() if __name__ == '__main__' and len(sys.argv) == 1: import world as wrld import car world = wrld.world2() vis = Visualizer(0.1, name='replay') vis.use_world(world) vis.main_car = world.cars[0] #vis.cars = [] #vis.cars.append(car.Car(world.cars[0].dyn,[-2., -2., 0., 0.], color='yellow')) vis.run() if __name__ == '__main__' and len(sys.argv) > 1: filename = sys.argv[1] import world world_name = (filename.split('/')[-1]).split('-')[0] magnify = 1. if len(sys.argv) > 3: magnify = float(sys.argv[3]) vis = Visualizer(0.2, name=world_name, magnify=magnify)
r = r - 30.*vis.lanes[2].shifted(-1).gaussian() r = r + 30.*feature.speed(0.5) r = r + 10.*vis.lanes[0].gaussian(10.) r = r + .1*feature.control() vis.cars[1].reward = r vis.main_car = vis.cars[0] vis.paused = True vis.set_heat(r) #vis.set_heat(vis.lanes[0].gaussian()+vis.lanes[1].gaussian()+vis.lanes[2].gaussian()) #vis.set_heat(-vis.cars[1].traj.gaussian()+vis.lanes[0].gaussian()+vis.lanes[1].gaussian()+vis.lanes[2].gaussian()) vis.run() if __name__ == '__main__' and len(sys.argv)==1: import world as wrld import car world = wrld.world2() vis = Visualizer(0.1, name='replay') vis.use_world(world) vis.main_car = world.cars[0] #vis.cars = [] #vis.cars.append(car.Car(world.cars[0].dyn,[-2., -2., 0., 0.], color='yellow')) vis.run() if __name__ == '__main__' and len(sys.argv)>1: filename = sys.argv[1] import world world_name = (filename.split('/')[-1]).split('-')[0] magnify = 1. if len(sys.argv)>3: magnify = float(sys.argv[3]) vis = Visualizer(0.2, name=world_name, magnify=magnify)