modify info

4 years ago · 814f707025
parent 51d885815a
commit 814f707025
1 changed files with 3 additions and 0 deletions
--- a/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc
+++ b/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc
@ -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++) {