huangfu/scenarionet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix nuplan bug

Browse Source
This commit is contained in:
QuanyiLi
2023-05-06 22:43:12 +01:00
parent 05c6844016
commit 300a29da09
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
scenarionet/converter/nuplan/utils.py
View File

@@ -50,11 +50,12 @@ def get_nuplan_scenarios(data_root, map_root, logs: Union[list, None] = None, bu
:return: :return:
""" """
nuplan_package_path = NUPLAN_PACKAGE_PATH nuplan_package_path = NUPLAN_PACKAGE_PATH
logs = logs or [file for file in os.listdir(data_root)] logs = logs or [file for file in os.listdir(data_root)]
log_string = "" log_string = ""
for log in logs: for log in logs:
log_string += log[:-3] # remove .db if log[-3:] == ".db":
log = log[:-3]
log_string += log
log_string += "," log_string += ","
log_string = log_string[:-1] log_string = log_string[:-1]
@@ -356,9 +357,9 @@ def extract_traffic(scenario: NuPlanScenario, center):
type=MetaDriveType.UNSET, type=MetaDriveType.UNSET,
state=dict( state=dict(
position=np.zeros(shape=(episode_len, 3)), position=np.zeros(shape=(episode_len, 3)),
heading=np.zeros(shape=(episode_len, )), heading=np.zeros(shape=(episode_len,)),
velocity=np.zeros(shape=(episode_len, 2)), velocity=np.zeros(shape=(episode_len, 2)),
valid=np.zeros(shape=(episode_len, )), valid=np.zeros(shape=(episode_len,)),
length=np.zeros(shape=(episode_len, 1)), length=np.zeros(shape=(episode_len, 1)),
width=np.zeros(shape=(episode_len, 1)), width=np.zeros(shape=(episode_len, 1)),
height=np.zeros(shape=(episode_len, 1)) height=np.zeros(shape=(episode_len, 1))