Searched refs:class_index_ (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | manifest_op.cc | 40 class_index_(class_index), in ManifestOp()
186 if (class_index_.empty() || class_index_.find(label_name) != class_index_.end()) { in ParseManifestFile()
252 label.second = class_index_.empty() ? index : class_index_[label.first]; in CountDatasetInfo()
305 if (!class_index_.empty()) { in GetClassIndexing()
307 … .emplace_back(std::make_pair(label.first, std::vector<int32_t>(1, class_index_[label.first]))); in GetClassIndexing()
|
| D | voc_op.cc | 56 class_index_(class_index), in VOCOp()
235 …if (label_name != "" && (class_index_.empty() || class_index_.find(label_name) != class_index_.end… in ParseAnnotationBbox()
290 if (class_index_.find(item.first) != class_index_.end()) { in ReadAnnotationToTensor()
291 label_data.push_back(static_cast<uint32_t>(class_index_[item.first])); in ReadAnnotationToTensor()
349 if (!class_index_.empty()) { in GetClassIndexing()
351 … .emplace_back(std::make_pair(label.first, std::vector<int32_t>(1, class_index_[label.first]))); in GetClassIndexing()
|
| D | image_folder_op.cc | 35 class_index_(map), in ImageFolderOp()
66 …v[ind]->second.front()->second = class_index_.empty() ? ind : class_index_[v[ind]->first.substr(1)… in PrescanMasterEntry()
189 if (class_index_.empty() || in RecursiveWalkFolder()
190 class_index_.find(subdir.ToString().substr(dirname_offset_ + 1)) != class_index_.end()) { in RecursiveWalkFolder()
316 …TURN_IF_NOT_OK(CountRowsAndClasses(folder_path_, extensions_, nullptr, num_classes, class_index_)); in GetNumClasses()
|
| D | manifest_op.h | 112 std::map<std::string, int32_t> class_index_; variable
|
| D | voc_op.h | 150 std::map<std::string, int32_t> class_index_; variable
|
| D | image_folder_op.h | 140 std::map<std::string, int32_t> class_index_; variable
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | voc_node.cc | 42 class_index_(class_indexing), in VOCNode()
50 …std::make_shared<VOCNode>(dataset_dir_, task_, usage_, class_index_, decode_, sampler, cache_, ext… in Copy()
65 if (!class_index_.empty()) { in ValidateParams()
126 …voc_op = std::make_shared<VOCOp>(task_type_, usage_, dataset_dir_, class_index_, num_workers_, con… in Build()
173 args["class_indexing"] = class_index_; in to_json()
|
| D | manifest_node.cc | 42 class_index_(class_indexing), in ManifestNode()
47 …auto node = std::make_shared<ManifestNode>(dataset_file_, usage_, sampler, class_index_, decode_, … in Copy()
105 …d::make_shared<ManifestOp>(num_workers_, dataset_file_, connector_que_size_, decode_, class_index_, in Build()
152 args["class_indexing"] = class_index_; in to_json()
|
| D | manifest_node.h | 77 const std::map<std::string, int32_t> &ClassIndex() const { return class_index_; } in ClassIndex()
104 std::map<std::string, int32_t> class_index_; variable
|
| D | voc_node.h | 78 const std::map<std::string, int32_t> &ClassIndex() const { return class_index_; } in ClassIndex()
112 std::map<std::string, int32_t> class_index_; variable
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | non_max_suppression_fp32.h | 56 int32_t class_index_; member
|