Fix the bug: tensorrt-cpp input blob from BGR to RGB Megvii-BaseDetec…

…tion#1758
vossr · Apr 15, 2024 · 2360277 · 2360277
1 parent 953037f
commit 2360277
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/demo/TensorRT/cpp/yolox.cpp b/demo/TensorRT/cpp/yolox.cpp
@@ -218,7 +218,8 @@ float* blobFromImage(cv::Mat& img){
         {
             for (size_t w = 0; w < img_w; w++) 
             {
-                blob[c * img_w * img_h + h * img_w + w] =
+                size_t invc = ((size_t)channels - 1) - c;
+                blob[invc * img_w * img_h + h * img_w + w] =
                     (float)img.at<cv::Vec3b>(h, w)[c];
             }
         }