From aaf41c67cce67c9bbe79f6935899532eccfe67ad Mon Sep 17 00:00:00 2001
From: triple-Mu <gpu@163.com>
Date: Sun, 19 Feb 2023 01:43:03 +0800
Subject: [PATCH] Fix line

---
 infer-seg.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/infer-seg.py b/infer-seg.py
index 1f3d3b2..46e3251 100644
--- 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(