def test(param):
sim = Simulator(640, 480, rate=int(1 / tau))
env = CartPoleEnv()
controller = Controller(param, 'net')
#controller = Controller(param,'pid')
s = env.reset()
g = CartPoleGraphics()
sim.add('cartpole', g)
data = {'s': s}
def step():
s = data['s']
ctrl = controller.compute(s)
s, fail, up = env.step(ctrl)
data['s'] = s
x, _, t, _ = s
g.update(x, t)
return fail
sim.run(step)
import sys
sys.path.insert(0, '../src/')
from simulator import Simulator
simulator = Simulator()
from colors import Yellow, White, DarkShadow
from layers.layers import Background, DrawLines, Perspective, Crop
white = White()
simulator.add(
Background(n_backgrounds=3, path='../ground_pics', input_size=(250, 200)))
simulator.add(DrawLines(input_size=(250, 200)))
simulator.add(Perspective())
simulator.add(Crop())
simulator.generate(n_examples=100, path='my_dataset')
