From e668de6e02e722a3cc546ccffe36a8c57e25e89f Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Fri, 29 Nov 2024 05:50:04 +0800
Subject: [PATCH] Add BGR to RGB conversion in LibTorch example (#17864)

---
 examples/YOLOv8-LibTorch-CPP-Inference/main.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/YOLOv8-LibTorch-CPP-Inference/main.cc b/examples/YOLOv8-LibTorch-CPP-Inference/main.cc
index b68b7f7e..0937b568 100644
--- a/examples/YOLOv8-LibTorch-CPP-Inference/main.cc
+++ b/examples/YOLOv8-LibTorch-CPP-Inference/main.cc
@@ -226,6 +226,7 @@ int main() {
         cv::Mat image = cv::imread("/path/to/bus.jpg");
         cv::Mat input_image;
         letterbox(image, input_image, {640, 640});
+        cv::cvtColor(input_image, input_image, cv::COLOR_BGR2RGB);
 
         torch::Tensor image_tensor = torch::from_blob(input_image.data, {input_image.rows, input_image.cols, 3}, torch::kByte).to(device);
         image_tensor = image_tensor.toType(torch::kFloat32).div(255);