File size: 484 Bytes
e516f1f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import h5py
def generate_map(
base_elevation,
obstacle_mask,
traffic_flow,
terrain_map,
output_path,
metadata={}
):
with h5py.File(output_path, 'w') as f:
f.create_dataset('elevation', data=base_elevation)
f.create_dataset('occupancy', data=obstacle_mask)
f.create_dataset('traffic_flow', data=traffic_flow)
f.create_dataset('terrain_type', data=terrain_map)
for k, v in metadata.items():
f.attrs[k] = v |