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

remove repeat calculation

Browse Source
This commit is contained in:
QuanyiLi
2023-05-21 14:45:01 +01:00
parent 1c01e4417e
commit 12132bc870
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scenarionet/converter/utils.py
View File

@@ -211,12 +211,8 @@ def write_to_directory_single_worker(
# add agents summary # add agents summary
summary_dict = {} summary_dict = {}
ego_car_id = sd_scenario[SD.METADATA][SD.SDC_ID]
summary_dict[ego_car_id] = SD.get_object_summary(
state_dict=sd_scenario.get_sdc_track()["state"], id=ego_car_id, type=sd_scenario.get_sdc_track()["type"]
)
for track_id, track in sd_scenario[SD.TRACKS].items(): for track_id, track in sd_scenario[SD.TRACKS].items():
summary_dict[track_id] = SD.get_object_summary(state_dict=track["state"], id=track_id, type=track["type"]) summary_dict[track_id] = SD.get_object_summary(state_dict=track, id=track_id)
sd_scenario[SD.METADATA][SD.SUMMARY.OBJECT_SUMMARY] = summary_dict sd_scenario[SD.METADATA][SD.SUMMARY.OBJECT_SUMMARY] = summary_dict
# count some objects occurrence # count some objects occurrence