Skip to content

Commit

Permalink
Fixed issue with erosion filter.
Browse files Browse the repository at this point in the history
  • Loading branch information
mktk1117 committed Apr 1, 2024
1 parent d33feca commit ee9a155
Show file tree
Hide file tree
Showing 3 changed files with 26 additions and 2 deletions.
3 changes: 2 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -8,4 +8,5 @@ docs/build
_build
.idea*
.vscode*
*.egg-info
*.egg-info
elevation_mapping_cupy/compile_commands.json
1 change: 0 additions & 1 deletion elevation_mapping_cupy/compile_commands.json

This file was deleted.

Original file line number Diff line number Diff line change
Expand Up @@ -28,13 +28,15 @@ def __init__(
kernel_size: int = 3,
iterations: int = 1,
reverse: bool = False,
default_layer_name: str = "traversability",
**kwargs,
):
super().__init__()
self.input_layer_name = input_layer_name
self.kernel_size = kernel_size
self.iterations = iterations
self.reverse = reverse
self.default_layer_name = default_layer_name

def __call__(
self,
Expand Down Expand Up @@ -69,6 +71,28 @@ def __call__(
semantic_layer_names,
self.input_layer_name,
)
if layer_data is None:
print(f"No layers are found, using {self.default_layer_name}!")
layer_data = self.get_layer_data(
elevation_map,
layer_names,
plugin_layers,
plugin_layer_names,
semantic_map,
semantic_layer_names,
self.default_layer_name,
)
if layer_data is None:
print(f"No layers are found, using traversability!")
layer_data = self.get_layer_data(
elevation_map,
layer_names,
plugin_layers,
plugin_layer_names,
semantic_map,
semantic_layer_names,
"traversability",
)
layer_np = cp.asnumpy(layer_data)

# Define the erosion kernel
Expand Down

0 comments on commit ee9a155

Please sign in to comment.