From f66d0affa3710c91c123c322ac1e3ace25d8a4de Mon Sep 17 00:00:00 2001
From: Allen Isaac Jose <allen.isaac@smail.inf.h-brs.de>
Date: Tue, 27 Feb 2024 16:03:40 +0100
Subject: [PATCH] Fix the bug: tensorrt-cpp input blob from BGR to RGB

---
 demo/TensorRT/cpp/yolox.cpp | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/demo/TensorRT/cpp/yolox.cpp b/demo/TensorRT/cpp/yolox.cpp
index ed423380e..e34e91b93 100644
--- a/demo/TensorRT/cpp/yolox.cpp
+++ b/demo/TensorRT/cpp/yolox.cpp
@@ -209,18 +209,19 @@ static void generate_yolox_proposals(std::vector<GridAndStride> grid_strides, fl
 
 float* blobFromImage(cv::Mat& img){
     float* blob = new float[img.total()*3];
-    int channels = 3;
     int img_h = img.rows;
     int img_w = img.cols;
-    for (size_t c = 0; c < channels; c++) 
-    {
-        for (size_t  h = 0; h < img_h; h++) 
-        {
-            for (size_t w = 0; w < img_w; w++) 
-            {
-                blob[c * img_w * img_h + h * img_w + w] =
-                    (float)img.at<cv::Vec3b>(h, w)[c];
-            }
+
+    int i = 0;
+	for (int row = 0; row < img_h; ++row) {
+		uchar* uc_pixel = img.data + row * img.step;
+		for (int col = 0; col < img_w; ++col)
+		{
+			blob[i] = (float)uc_pixel[2];
+			blob[i + img_h * img_w] = (float)uc_pixel[1];
+			data[i + 2 * img_h * img_w] = (float)uc_pixel[0];
+			uc_pixel += 3;
+			++i;
         }
     }
     return blob;