fix normalization to_rgb option (#1771)

open-mmlab · Feb 21, 2023 · c941045 · c941045
1 parent 4bb8920
commit c941045
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/csrc/mmdeploy/preprocess/transform/normalize.cpp b/csrc/mmdeploy/preprocess/transform/normalize.cpp
@@ -102,8 +102,8 @@ class Normalize : public Transform {
       } else if (to_rgb_) {
         auto src_mat = to_mat(tensor, PixelFormat::kBGR);
         Mat dst_mat;
-        OUTCOME_TRY(cvt_color_.Apply(src_mat, dst_mat, PixelFormat::kBGR));
-        dst = to_tensor(src_mat);
+        OUTCOME_TRY(cvt_color_.Apply(src_mat, dst_mat, PixelFormat::kRGB));
+        dst = to_tensor(dst_mat);
       }
       data[key] = std::move(dst);