Merge branch 'main' of github.com:meituan/YOLOv6

SangbumChoi · Mar 24, 2023 · 9cfd42c · 9cfd42c
2 parents 2350da2 + ae7d17c
commit 9cfd42c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",