Fix line

2 years ago · aaf41c67cc
parent 7902783618
commit aaf41c67cc
1 changed files with 1 additions and 0 deletions
--- a/infer-seg.py
+++ b/infer-seg.py
@ -37,6 +37,7 @@ def main(args: argparse.Namespace) -> None:
        tensor = torch.asarray(tensor, device=device)
        # inference
        data = Engine(tensor)
+
        seg_img = torch.asarray(seg_img[dh:H - dh, dw:W - dw, [2, 1, 0]],
                                device=device)
        bboxes, scores, labels, masks = seg_postprocess(