Browse Source

!9353 Add check to config.num_default.size()

From: @shenwei41
Reviewed-by: @pandoublefeng,@heleiwang
Signed-off-by: @pandoublefeng
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
b5aec1fd45
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc

+ 3
- 0
mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc View File

@@ -747,6 +747,9 @@ std::vector<std::vector<float>> GetDefaultBoxes(BoxesConfig config) {
for (int i = 0; i < config.steps.size(); i++) {
fk.push_back(num / config.steps[i]);
}
if (config.num_default.size() < 2) {
return {};
}
float scale_rate = (config.max_scale - config.min_scale) / (config.num_default.size() - 1);
std::vector<float> scales(config.num_default.size());
for (int i = 0; i < scales.size(); i++) {


Loading…
Cancel
Save