diff --git a/src/universal_devkit/scripts/supervisely_3d_to_universal.py b/src/universal_devkit/scripts/supervisely_3d_to_universal.py index fca0515..7800961 100644 --- a/src/universal_devkit/scripts/supervisely_3d_to_universal.py +++ b/src/universal_devkit/scripts/supervisely_3d_to_universal.py @@ -23,6 +23,12 @@ def convert_supervisely_3d_to_universal(input_directory, output_directory): input_data = read_json(filepath) + objectkey2class = {} + for object in input_data["objects"]: + object_key = object["key"] + class_label = object["classTitle"] + objectkey2class[object_key] = class_label + output_data = [] for figure in input_data["figures"]: ann = {} @@ -52,6 +58,8 @@ def convert_supervisely_3d_to_universal(input_directory, output_directory): figure_geometry["rotation"]["z"], ] + ann["class_label"] = objectkey2class[figure["objectKey"]] + ann["prev"] = "" ann["next"] = "" ann["num_lidar"] = 0