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