From 814f7070253320949465ca061040b709d3f6d985 Mon Sep 17 00:00:00 2001
From: shenwei41 <shenwei41@huawei.com>
Date: Wed, 2 Dec 2020 10:57:52 +0800
Subject: [PATCH] modify info

---
 .../minddata/dataset/kernels/image/lite_cv/image_process.cc    | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc b/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.cc
index 59296c885a..7681b9f73d 100644
--- 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++) {