diff --git a/inference.ipynb b/inference.ipynb index 4b3ad409..644f2372 100644 --- a/inference.ipynb +++ b/inference.ipynb @@ -120,7 +120,7 @@ " image = letterbox(img_src, img_size, stride=stride)[0]\n", "\n", " # Convert\n", - " image = image.transpose((2, 0, 1))[::-1] # HWC to CHW, RGB to BGR\n", + " image = image.transpose((2, 0, 1)) # HWC to CHW\n", " image = torch.from_numpy(np.ascontiguousarray(image))\n", " image = image.half() if half else image.float() # uint8 to fp16/32\n", " image /= 255 # 0 - 255 to 0.0 - 1.0\n",