From 10846400080a46533e36f7dd599477105ffcd9f2 Mon Sep 17 00:00:00 2001 From: ktro2828 Date: Fri, 6 Sep 2024 15:27:40 +0900 Subject: [PATCH] chode: fix typo of `occlusion` Signed-off-by: ktro2828 --- perception_dataset/deepen/deepen_to_t4_converter.py | 6 +++--- .../fastlabel_to_t4/fastlabel_2d_to_t4_converter.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/perception_dataset/deepen/deepen_to_t4_converter.py b/perception_dataset/deepen/deepen_to_t4_converter.py index dd288bc3..cd351a99 100644 --- a/perception_dataset/deepen/deepen_to_t4_converter.py +++ b/perception_dataset/deepen/deepen_to_t4_converter.py @@ -127,7 +127,7 @@ def _make_rosbag(self, input_bag_dir: str, output_dir: str): converter.convert() shutil.move(output_bag_dir_temp, output_bag_dir) - def _convert_occulusion_to_visibility(self, name: str) -> str: + def _convert_occlusion_to_visibility(self, name: str) -> str: if name == "full": return "none" elif name == "partial": @@ -213,11 +213,11 @@ def _format_deepen_annotation( else: anno_attributes: Dict[str, str] = label_dict["attributes"] if "Occlusion_State" in anno_attributes: - visibility: str = self._convert_occulusion_to_visibility( + visibility: str = self._convert_occlusion_to_visibility( anno_attributes["Occlusion_State"] ) elif "occlusion_state" in anno_attributes: - visibility: str = self._convert_occulusion_to_visibility( + visibility: str = self._convert_occlusion_to_visibility( anno_attributes["occlusion_state"] ) else: diff --git a/perception_dataset/fastlabel_to_t4/fastlabel_2d_to_t4_converter.py b/perception_dataset/fastlabel_to_t4/fastlabel_2d_to_t4_converter.py index a9d11da8..1f352e2f 100644 --- a/perception_dataset/fastlabel_to_t4/fastlabel_2d_to_t4_converter.py +++ b/perception_dataset/fastlabel_to_t4/fastlabel_2d_to_t4_converter.py @@ -189,7 +189,7 @@ def _format_fastlabel_annotation(self, annotations: Dict[str, List[Dict[str, Any occlusion_state = ( "occlusion_state." + att["key"].split("_")[-1] ) - visibility = self._convert_occulusion_to_visibility( + visibility = self._convert_occlusion_to_visibility( att["key"].split("_")[-1] ) break