* fix a bug in csrc/mmdeploy/preprocess/transform/normalize.cpp

open-mmlab · Dec 24, 2022 · e6a21d3 · e6a21d3
1 parent 7bcac39
commit e6a21d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/csrc/mmdeploy/preprocess/transform/normalize.cpp b/csrc/mmdeploy/preprocess/transform/normalize.cpp
@@ -99,13 +99,14 @@ class Normalize : public Transform {
       Tensor dst;
       if (to_float_) {
         OUTCOME_TRY(normalize_.Apply(tensor, dst));
+        data[key] = std::move(dst);
       } 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);
+        data[key] = std::move(dst);
       }
-      data[key] = std::move(dst);
 
       for (auto& v : mean_) {
         data["img_norm_cfg"]["mean"].push_back(v);