del CR in VOC txt

5 years ago · 9e48e3daaa
parent 69edd8e2fc
commit 9e48e3daaa
1 changed files with 5 additions and 1 deletions
--- a/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc
+++ b/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc
@ -270,8 +270,12 @@ Status VOCOp::ParseImageIds() {
  }
  std::string id;
  while (getline(in_file, id)) {
+    if (id[id.size() - 1] == '\r') {
+      image_ids_.push_back(id.substr(0, id.size() - 1));
+    } else {
      image_ids_.push_back(id);
    }
+  }
  in_file.close();
  image_ids_.shrink_to_fit();
  num_rows_ = image_ids_.size();