浏览代码

add hiai_cv_saliencyDetectorModel.tflite

fix space_to_batch bug
tags/v1.0.0
jianghui58 5 年前
父节点
当前提交
20de8b6d9b
共有 2 个文件被更改,包括 2 次插入1 次删除
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc
  2. +1
    -0
      mindspore/lite/test/models_tflite.cfg

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp32/space_to_batch.cc 查看文件

@@ -89,7 +89,7 @@ int SpaceToBatchCPUKernel::ReSize() {
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 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));
if (pedding_h_data_ == nullptr) {
MS_LOG(ERROR) << "malloc pedding h data fail!";


+ 1
- 0
mindspore/lite/test/models_tflite.cfg 查看文件

@@ -89,6 +89,7 @@ hiai_cv_labelDetectorModel_v2.tflite
#hiai_cv_labelDetectorModel_v3.tflite
hiai_cv_labelDetectorModel_v4.tflite
hiai_dress_detect.tflite
hiai_cv_saliencyDetectorModel.tflite
hiai_frozen_inference_graph.tflite
hiai_ghostnet.tflite
hiai_iMaxDN_RGB.tflite


正在加载...
取消
保存