Home
last modified time | relevance | path

Searched refs:CocoNode (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dcoco_node.cc34 CocoNode::CocoNode(const std::string &dataset_dir, const std::string &annotation_file, const std::s… in CocoNode() function in mindspore::dataset::CocoNode
45 std::shared_ptr<DatasetNode> CocoNode::Copy() { in Copy()
48 …std::make_shared<CocoNode>(dataset_dir_, annotation_file_, task_, decode_, sampler, cache_, extra_… in Copy()
52 void CocoNode::Print(std::ostream &out) const { out << Name(); } in Print()
54 Status CocoNode::ValidateParams() { in ValidateParams()
78 Status CocoNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
153 Status CocoNode::GetShardId(int32_t *shard_id) { in GetShardId()
160 Status CocoNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool estimat… in GetDatasetSize()
183 Status CocoNode::to_json(nlohmann::json *out_json) { in to_json()
203 Status CocoNode::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetNode> *ds) { in from_json()
[all …]
Dcoco_node.h29 class CocoNode : public MappableSourceNode {
32CocoNode(const std::string &dataset_dir, const std::string &annotation_file, const std::string &ta…
37 ~CocoNode() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc117 PYBIND_REGISTER(CocoNode, 2, ([](const py::module *m) { in __anon810239690b02()
118 (void)py::class_<CocoNode, DatasetNode, std::shared_ptr<CocoNode>>(*m, "CocoNode", in __anon810239690b02()
122 std::shared_ptr<CocoNode> coco = std::make_shared<CocoNode>( in __anon810239690b02()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h63 class CocoNode; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dserdes.cc160 RETURN_IF_NOT_OK(CocoNode::from_json(json_obj, ds)); in CreateDatasetNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc985 …auto ds = std::make_shared<CocoNode>(CharToString(dataset_dir), CharToString(annotation_file), Cha… in CocoDataset()
993 …auto ds = std::make_shared<CocoNode>(CharToString(dataset_dir), CharToString(annotation_file), Cha… in CocoDataset()
1002 …auto ds = std::make_shared<CocoNode>(CharToString(dataset_dir), CharToString(annotation_file), Cha… in CocoDataset()
/third_party/mindspore/tests/ut/cpp/dataset/
Ddeserialize_test.cc375 std::make_shared<CocoNode>(folder_path, annotation_file, task, true, sampler, cache, false); in TEST_F()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py5323 … return cde.CocoNode(self.dataset_dir, self.annotation_file, self.task, self.decode, self.sampler,