Skip to content

Commit

Permalink
chode: fix typo of occlusion (#151)
Browse files Browse the repository at this point in the history
Signed-off-by: ktro2828 <[email protected]>
  • Loading branch information
ktro2828 authored Sep 6, 2024
1 parent ac556fd commit 76286a6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions perception_dataset/deepen/deepen_to_t4_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -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":
Expand Down Expand Up @@ -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:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 76286a6

Please sign in to comment.