Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "deef9a3df883b76caeca7d35ccbe960a66af29de"
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,
for _idx in instances:
instance_idx = (_seg == _idx).nonzero(as_tuple=False)
_mins = instance_idx[:, -3:].min(dim=0)[0]
_maxs = instance_idx[:, -3:].max(dim=0)[0]
_mins = instance_idx[:, -dim:].min(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]
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