p83571920
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

From: @shenwei41
Reviewed-by: @pandoublefeng,@heleiwang
Signed-off-by: @pandoublefeng
This commit is contained in:
mindspore-ci-bot 2020-12-02 14:21:25 +08:00 committed by Gitee
parent ca053f0cbc 814f707025
commit b5aec1fd45
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
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++) {