Skip to content

GitLab

Commit 4adbed5d authored by mibaumgartner's avatar mibaumgartner
Browse files

instances_to_boxes 2d support

parent ea565ddd
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nndet/io/transforms/instances.py
View file @ 4adbed5d
...@@ -120,8 +120,8 @@ def instances_to_boxes(seg: Tensor, ...@@ -120,8 +120,8 @@ def instances_to_boxes(seg: Tensor,
for _idx in instances: for _idx in instances:
instance_idx = (_seg == _idx).nonzero(as_tuple=False) instance_idx = (_seg == _idx).nonzero(as_tuple=False)
_mins = instance_idx[:, -3:].min(dim=0)[0] _mins = instance_idx[:, -dim:].min(dim=0)[0]
_maxs = instance_idx[:, -3:].max(dim=0)[0] _maxs = instance_idx[:, -dim:].max(dim=0)[0]
box = [_mins[-dim] - 1, _mins[(-dim) + 1] - 1, _maxs[-dim] + 1, _maxs[(-dim) + 1] + 1] box = [_mins[-dim] - 1, _mins[(-dim) + 1] - 1, _maxs[-dim] + 1, _maxs[(-dim) + 1] + 1]
if dim > 2: if dim > 2:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment