From 4c70d25777717834d02c231810fc3b636f32eb08 Mon Sep 17 00:00:00 2001
From: xiefangqi <xiefangqi2@huawei.com>
Date: Sat, 30 May 2020 15:07:36 +0800
Subject: [PATCH] fix codex of voc and graph

---
 mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc | 5 +++--
 mindspore/ccsrc/dataset/engine/gnn/graph_loader.h          | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc b/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc
index a95f1fe61a..d96b3a8872 100644
--- a/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc
+++ b/mindspore/ccsrc/dataset/engine/datasetops/source/voc_op.cc
@@ -49,6 +49,7 @@ VOCOp::Builder::Builder() : builder_decode_(false), builder_num_samples_(0), bui
   builder_num_workers_ = cfg->num_parallel_workers();
   builder_rows_per_buffer_ = cfg->rows_per_buffer();
   builder_op_connector_size_ = cfg->op_connector_size();
+  builder_task_type_ = TaskType::Segmentation;
 }
 
 Status VOCOp::Builder::Build(std::shared_ptr<VOCOp> *ptr) {
@@ -270,7 +271,7 @@ Status VOCOp::ParseImageIds() {
   }
   std::string id;
   while (getline(in_file, id)) {
-    if (id[id.size() - 1] == '\r') {
+    if (id.size() > 0 && id[id.size() - 1] == '\r') {
       image_ids_.push_back(id.substr(0, id.size() - 1));
     } else {
       image_ids_.push_back(id);
@@ -330,7 +331,7 @@ Status VOCOp::ParseAnnotationBbox(const std::string &path) {
     std::string label_name;
     uint32_t xmin = 0, ymin = 0, xmax = 0, ymax = 0, truncated = 0, difficult = 0;
     XMLElement *name_node = object->FirstChildElement("name");
-    if (name_node != nullptr) label_name = name_node->GetText();
+    if (name_node != nullptr && name_node->GetText() != 0) label_name = name_node->GetText();
     XMLElement *truncated_node = object->FirstChildElement("truncated");
     if (truncated_node != nullptr) truncated = truncated_node->UnsignedText();
     XMLElement *difficult_node = object->FirstChildElement("difficult");
diff --git a/mindspore/ccsrc/dataset/engine/gnn/graph_loader.h b/mindspore/ccsrc/dataset/engine/gnn/graph_loader.h
index 7e5ccdd35f..0ad54bae6d 100644
--- a/mindspore/ccsrc/dataset/engine/gnn/graph_loader.h
+++ b/mindspore/ccsrc/dataset/engine/gnn/graph_loader.h
@@ -53,6 +53,7 @@ class GraphLoader {
  public:
   explicit GraphLoader(std::string mr_filepath, int32_t num_workers = 4);
 
+  ~GraphLoader() = default;
   // Init mindrecord and load everything into memory multi-threaded
   // @return Status - the status code
   Status InitAndLoad();