NVlabs / trajdata

A unified interface to many trajectory forecasting datasets.
Apache License 2.0
311 stars 38 forks source link

ValueError when caching Waymo dataset (float NaN conversion issue for vec_map) #29

Closed jxmmy7777 closed 1 year ago

jxmmy7777 commented 1 year ago

Hi everyone,

I'm encountering an issue when trying to cache the Waymo dataset. Although caching works as expected for the Lyft dataset, I experienced a ValueError with Waymo.

When I attempt to cache the Waymo dataset, I run into an error related to converting a float NaN value to an integer for vec_map.

File "/path/to/trajdata/examples/batch_example.py", line 55, in <module>
    main()
  File "/path/to/trajdata/examples/batch_example.py", line 14, in main
    dataset = UnifiedDataset(
  ... [truncated for brevity]
  File "/path/to/trajdata/src/trajdata/utils/raster_utils.py", line 145, in rasterize_map
    raster_size_x: int = ceil((max_x - min_x) * resolution)
ValueError: cannot convert float NaN to integer 

The vec_map.extent is an array of NaN Would appreciate any help or guidance on this. Thank you!