From d3bdf0e1fa0cf19266187a92cb0f612b4d0b20ea Mon Sep 17 00:00:00 2001 From: Elijah Harmon Date: Fri, 2 Sep 2022 16:31:40 -0400 Subject: [PATCH] Fixed indent issue --- main_tensorrt_gpu.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/main_tensorrt_gpu.py b/main_tensorrt_gpu.py index cb18040..c9c9d46 100644 --- a/main_tensorrt_gpu.py +++ b/main_tensorrt_gpu.py @@ -98,9 +98,9 @@ def main(): im = torch.from_numpy(cp.asnumpy(im)).to('cuda') #Converting to numpy for visuals - im0 = im[0].permute(1, 2, 0) * 255 - im0 = im0.cpu().numpy().astype(np.uint8) - im0 = cv2.cvtColor(im0, cv2.COLOR_RGB2BGR) #Image has to be in BGR for visualization + im0 = im[0].permute(1, 2, 0) * 255 + im0 = im0.cpu().numpy().astype(np.uint8) + im0 = cv2.cvtColor(im0, cv2.COLOR_RGB2BGR) #Image has to be in BGR for visualization # Detecting all the objects results = model(im)