提交 20de8b6d 编写于 作者: J jianghui58

add hiai_cv_saliencyDetectorModel.tflite

fix space_to_batch bug
上级 8d419314
...@@ -89,7 +89,7 @@ int SpaceToBatchCPUKernel::ReSize() { ...@@ -89,7 +89,7 @@ int SpaceToBatchCPUKernel::ReSize() {
padded_in_shape_[2] = in_shape[2] + param->paddings_[2] + param->paddings_[3]; padded_in_shape_[2] = in_shape[2] + param->paddings_[2] + param->paddings_[3];
auto num_elements_padded = EnumElement(padded_in_shape_.data(), in_shape.size()); auto num_elements_padded = EnumElement(padded_in_shape_.data(), in_shape.size());
auto output_shape = out_tensors_[0]->shape(); auto output_shape = out_tensors_[0]->shape();
auto pedding_h_size = output_shape[2] * output_shape[3] * sizeof(float); auto pedding_h_size = padded_in_shape_[2] * output_shape[3] * sizeof(float);
pedding_h_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(pedding_h_size)); pedding_h_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(pedding_h_size));
if (pedding_h_data_ == nullptr) { if (pedding_h_data_ == nullptr) {
MS_LOG(ERROR) << "malloc pedding h data fail!"; MS_LOG(ERROR) << "malloc pedding h data fail!";
......
...@@ -89,6 +89,7 @@ hiai_cv_labelDetectorModel_v2.tflite ...@@ -89,6 +89,7 @@ hiai_cv_labelDetectorModel_v2.tflite
#hiai_cv_labelDetectorModel_v3.tflite #hiai_cv_labelDetectorModel_v3.tflite
hiai_cv_labelDetectorModel_v4.tflite hiai_cv_labelDetectorModel_v4.tflite
hiai_dress_detect.tflite hiai_dress_detect.tflite
hiai_cv_saliencyDetectorModel.tflite
hiai_frozen_inference_graph.tflite hiai_frozen_inference_graph.tflite
hiai_ghostnet.tflite hiai_ghostnet.tflite
hiai_iMaxDN_RGB.tflite hiai_iMaxDN_RGB.tflite
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册