From d6d5ad8ba7f1c3a81f70a656fa3a806f6a63b231 Mon Sep 17 00:00:00 2001 From: Yingchen Ma Date: Sat, 4 Sep 2021 20:50:45 -0400 Subject: [PATCH 1/2] FEAT: add class attribute to supervisely_3d_to_universal --- .../scripts/supervisely_3d_to_universal.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/universal_devkit/scripts/supervisely_3d_to_universal.py b/src/universal_devkit/scripts/supervisely_3d_to_universal.py index fca0515..52d2944 100644 --- a/src/universal_devkit/scripts/supervisely_3d_to_universal.py +++ b/src/universal_devkit/scripts/supervisely_3d_to_universal.py @@ -18,11 +18,16 @@ def convert_supervisely_3d_to_universal(input_directory, output_directory): Path(output_directory).mkdir(parents=True, exist_ok=True) for filepath in glob("{}/*.pcd.json".format(input_directory)): - # Get the filename with extensions (ex. 1212129.pcd.json) filename = Path(filepath).name 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 +57,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 From 1a4a4c40975b274696627e391a0287f236888d8f Mon Sep 17 00:00:00 2001 From: Yingchen Ma Date: Sat, 4 Sep 2021 20:53:46 -0400 Subject: [PATCH 2/2] STYLE: re-add missing comment --- src/universal_devkit/scripts/supervisely_3d_to_universal.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/universal_devkit/scripts/supervisely_3d_to_universal.py b/src/universal_devkit/scripts/supervisely_3d_to_universal.py index 52d2944..7800961 100644 --- a/src/universal_devkit/scripts/supervisely_3d_to_universal.py +++ b/src/universal_devkit/scripts/supervisely_3d_to_universal.py @@ -18,6 +18,7 @@ def convert_supervisely_3d_to_universal(input_directory, output_directory): Path(output_directory).mkdir(parents=True, exist_ok=True) for filepath in glob("{}/*.pcd.json".format(input_directory)): + # Get the filename with extensions (ex. 1212129.pcd.json) filename = Path(filepath).name input_data = read_json(filepath)