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

batch convert nuscenes

Browse Source
This commit is contained in:
QuanyiLi
2023-05-06 17:10:12 +01:00
parent 8eb2e07d50
commit 311932c986
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scenarionet/converter/utils.py
View File

@@ -60,7 +60,7 @@ def mph_to_kmh(speed_in_mph: float):
def get_agent_summary(state_dict, id, type):
track = state_dict["position"]
valid_track = track[state_dict["valid"], :2]
valid_track = track[state_dict["valid"].astype(int), :2]
distance = float(sum(np.linalg.norm(valid_track[i] - valid_track[i + 1]) for i in range(valid_track.shape[0] - 1)))
valid_length = int(sum(state_dict["valid"]))
@@ -173,10 +173,11 @@ def write_to_directory(convert_func,
# rename and save
if delay_remove is not None:
assert delay_remove == save_path
shutil.rmtree(delay_remove)
os.rename(output_path, save_path)
summary_file = os.path.join(save_path, summary_file)
with open(summary_file, "wb") as file:
pickle.dump(dict_recursive_remove_array(metadata_recorder), file)
print("Summary is saved at: {}".format(summary_file))
assert delay_remove == save_path