Metadrive: fix map config on main (#30983)

* Fix map api too

* dont render
pull/30984/head
Justin Newberry 1 year ago committed by GitHub
parent 3d73512c07
commit 28e5d1bb49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 34
      tools/sim/bridge/metadrive/metadrive_bridge.py

@ -60,6 +60,24 @@ def curve_block(length, angle=45, direction=0):
"dir": direction "dir": direction
} }
def create_map(track_size=60):
return dict(
type=MapGenerateMethod.PG_MAP_FILE,
lane_num=2,
lane_width=3.5,
config=[
None,
straight_block(track_size),
curve_block(track_size*2, 90),
straight_block(track_size),
curve_block(track_size*2, 90),
straight_block(track_size),
curve_block(track_size*2, 90),
straight_block(track_size),
curve_block(track_size*2, 90),
]
)
class MetaDriveBridge(SimulatorBridge): class MetaDriveBridge(SimulatorBridge):
TICKS_PER_FRAME = 5 TICKS_PER_FRAME = 5
@ -82,7 +100,6 @@ class MetaDriveBridge(SimulatorBridge):
vehicle_config=dict( vehicle_config=dict(
enable_reverse=False, enable_reverse=False,
image_source="rgb_road", image_source="rgb_road",
spawn_longitude=15
), ),
sensors=sensors, sensors=sensors,
image_on_cuda=_cuda_enable, image_on_cuda=_cuda_enable,
@ -93,20 +110,7 @@ class MetaDriveBridge(SimulatorBridge):
crash_vehicle_done=False, crash_vehicle_done=False,
crash_object_done=False, crash_object_done=False,
traffic_density=0.0, # traffic is incredibly expensive traffic_density=0.0, # traffic is incredibly expensive
map_config=dict( map_config=create_map(),
type=MapGenerateMethod.PG_MAP_FILE,
config=[
None,
straight_block(120),
curve_block(240, 90),
straight_block(120),
curve_block(240, 90),
straight_block(120),
curve_block(240, 90),
straight_block(120),
curve_block(240, 90),
]
),
decision_repeat=1, decision_repeat=1,
physics_world_step_size=self.TICKS_PER_FRAME/100, physics_world_step_size=self.TICKS_PER_FRAME/100,
preload_models=False preload_models=False

Loading…
Cancel
Save