Browse Source

!2053 add check if there is no data

Merge pull request !2053 from yanghaitao/yht_no_data
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
a2a90800a0
2 changed files with 10 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/dataset/engine/datasetops/source/image_folder_op.cc
  2. +5
    -0
      mindspore/ccsrc/dataset/engine/datasetops/source/mnist_op.cc

+ 5
- 0
mindspore/ccsrc/dataset/engine/datasetops/source/image_folder_op.cc View File

@@ -116,6 +116,11 @@ Status ImageFolderOp::PrescanMasterEntry(const std::string &filedir) {
}
image_label_pairs_.shrink_to_fit();
num_rows_ = image_label_pairs_.size();
if (num_rows_ == 0) {
RETURN_STATUS_UNEXPECTED(
"There is no valid data matching the dataset API ImageFolderDatasetV2.Please check file path or dataset "
"API validation first.");
}
// free memory of two queues used for pre-scan
folder_name_queue_->Reset();
image_name_queue_->Reset();


+ 5
- 0
mindspore/ccsrc/dataset/engine/datasetops/source/mnist_op.cc View File

@@ -350,6 +350,11 @@ Status MnistOp::ParseMnistData() {
}
image_label_pairs_.shrink_to_fit();
num_rows_ = image_label_pairs_.size();
if (num_rows_ == 0) {
RETURN_STATUS_UNEXPECTED(
"There is no valid data matching the dataset API MnistDataset.Please check file path or dataset API "
"validation first.");
}
return Status::OK();
}



Loading…
Cancel
Save