diff --git a/examples/demo.py b/examples/demo.py index c77dc814b23e4402cc8cd52fc7e20c9528f4bd6b..c40cebfc0805838e628324548831ff97868514d9 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -131,10 +131,10 @@ class Demo: self.state_size = 105 * 2 self.action_size = 4 self.agent = Agent(self.state_size, self.action_size, "FC", 0) - self.agent.qnetwork_local.load_state_dict(torch.load('../flatland/baselines/Nets/avoid_checkpoint15000.pth')) + self.agent.qnetwork_local.load_state_dict(torch.load('./flatland/baselines/Nets/avoid_checkpoint15000.pth')) def create_renderer(self): - self.renderer = RenderTool(self.env, gl="QT") + self.renderer = RenderTool(self.env, gl="QTSVG") handle = self.env.get_agent_handles() return handle @@ -201,14 +201,14 @@ if False: demo_001.run_demo() demo_001 = None -demo_000 = Demo(Scenario_Generator.load_scenario('../env-data/railway/example_network_000.pkl')) +demo_000 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_000.pkl')) demo_000.run_demo() demo_000 = None -demo_001 = Demo(Scenario_Generator.load_scenario('../env-data/railway/example_network_001.pkl')) +demo_001 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_001.pkl')) demo_001.run_demo() demo_001 = None -demo_002 = Demo(Scenario_Generator.load_scenario('../env-data/railway/example_network_002.pkl')) +demo_002 = Demo(Scenario_Generator.load_scenario('./env-data/railway/example_network_002.pkl')) demo_002.run_demo() demo_002 = None diff --git a/flatland/utils/render_qt.py b/flatland/utils/render_qt.py index ea9613968c473c351849509bfc3e277dd7fe0701..2a4a02a78a1dbefe406f538c277ed7f91bc08094 100644 --- a/flatland/utils/render_qt.py +++ b/flatland/utils/render_qt.py @@ -170,6 +170,7 @@ class QTSVG(GraphicsLayer): def setRailAt(self, row, col, binTrans): if binTrans in self.track.dSvg: sSVG = self.track.dSvg[binTrans].to_string() + sSVG = sSVG.replace("ASCII", "UTF-8") bySVG = bytearray(sSVG, encoding='utf-8') svgWidget = QtSvg.QSvgWidget() svgWidget.renderer().load(bySVG)