Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/
Dimage_folder_node.cc35 ImageFolderNode::ImageFolderNode(std::string dataset_dir, bool decode, std::shared_ptr<SamplerObj> … in ImageFolderNode() function in mindspore::dataset::ImageFolderNode
47 std::shared_ptr<DatasetNode> ImageFolderNode::Copy() { in Copy()
50 …std::make_shared<ImageFolderNode>(dataset_dir_, decode_, sampler, recursive_, exts_, class_indexin… in Copy()
54 void ImageFolderNode::Print(std::ostream &out) const { in Print()
58 Status ImageFolderNode::ValidateParams() { in ValidateParams()
67 Status ImageFolderNode::Build(std::vector<std::shared_ptr<DatasetOp>> *const node_ops) { in Build()
87 Status ImageFolderNode::GetShardId(int32_t *shard_id) { in GetShardId()
94 Status ImageFolderNode::GetDatasetSize(const std::shared_ptr<DatasetSizeGetter> &size_getter, bool … in GetDatasetSize()
113 Status ImageFolderNode::to_json(nlohmann::json *out_json) { in to_json()
133 Status ImageFolderNode::from_json(nlohmann::json json_obj, std::shared_ptr<DatasetNode> *ds) { in from_json()
[all …]
Dimage_folder_node.h34 class ImageFolderNode : public MappableSourceNode {
37ImageFolderNode(std::string dataset_dir, bool decode, std::shared_ptr<SamplerObj> sampler, bool re…
42 ~ImageFolderNode() = default;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/engine/ir/datasetops/source/
Dbindings.cc186 PYBIND_REGISTER(ImageFolderNode, 2, ([](const py::module *m) { in __anon810239691602()
187 (void)py::class_<ImageFolderNode, DatasetNode, std::shared_ptr<ImageFolderNode>>( in __anon810239691602()
193 … auto imagefolder = std::make_shared<ImageFolderNode>(dataset_dir, decode, toSamplerObj(sampler), in __anon810239691602()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/opt/
Dpass.h64 class ImageFolderNode; variable
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/
Dserdes.cc166 RETURN_IF_NOT_OK(ImageFolderNode::from_json(json_obj, ds)); in CreateDatasetNode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Ddatasets.cc1080 …auto ds = std::make_shared<ImageFolderNode>(CharToString(dataset_dir), decode, sampler_obj, recurs… in ImageFolderDataset()
1094 …auto ds = std::make_shared<ImageFolderNode>(CharToString(dataset_dir), decode, sampler_obj, recurs… in ImageFolderDataset()
1109 …auto ds = std::make_shared<ImageFolderNode>(CharToString(dataset_dir), decode, sampler_obj, recurs… in ImageFolderDataset()
/third_party/mindspore/tests/ut/cpp/dataset/
Ddeserialize_test.cc278 …std::make_shared<ImageFolderNode>(dataset_dir, false, sampler, false, extensions, class_indexing, … in TEST_F()
/third_party/mindspore/mindspore/dataset/engine/
Ddatasets.py3391 …return cde.ImageFolderNode(self.dataset_dir, self.decode, self.sampler, self.extensions, self.clas…