leon 1 місяць тому
батько
коміт
8158ab4450
1 змінених файлів з 3 додано та 1 видалено
  1. 3 1
      src/infer/trt/depth/depth.cu

+ 3 - 1
src/infer/trt/depth/depth.cu

@@ -20,7 +20,9 @@ bool DepthModelImpl::load(const std::string &engine_file)
     network_input_width_ = input_dim[3];
     network_input_height_ = input_dim[2];
     isdynamic_model_ = trt_->has_dynamic_dim();
-    normalize_ = affine::Norm::mean_std({0.485f, 0.456f, 0.406f}, {0.229f, 0.224f, 0.225f}, 1 / 255.0f, affine::ChannelType::SwapRB);
+    float mean[3] = {0.485f, 0.456f, 0.406f};
+    float std[3] = {0.229f, 0.224f, 0.225f};
+    normalize_ = affine::Norm::mean_std(mean, std, 1 / 255.0f, affine::ChannelType::SwapRB);
     // normalize_ = affine::Norm::alpha_beta(1 / 255.0f, 0.0f, affine::ChannelType::SwapRB);
     return true;
 }