Home
last modified time | relevance | path

Searched refs:UniformAugment (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/tests/ut/cpp/dataset/
Dc_api_vision_uniform_aug_test.cc48 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
96 auto uniform_aug_op(new vision::UniformAugment({random_crop_op, center_crop_op}, 2)); in TEST_F()
149 …vision::UniformAugment uniform_aug_op = vision::UniformAugment({random_crop_op, center_crop_op}, 2… in TEST_F()
191 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
216 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
241 std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op}, 1)); in TEST_F()
265 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, nullpt… in TEST_F()
287 std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment(list, 1)); in TEST_F()
Dexecute_test.cc441 …std::shared_ptr<TensorTransform> uniform_op(new vision::UniformAugment({resize_op, vertical, horiz… in TEST_F()
/third_party/mindspore/tests/ut/python/dataset/
Dtest_uniform_augment.py45 uni_aug = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
88 … F.UniformAugment(transforms=transform_list,
146 uni_aug = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
187 C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
206 _ = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
226 _ = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
246 _ = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
267 uni_aug = C.UniformAugment(transforms=transforms_ua, num_ops=num_ops)
Dtest_serdes_dataset.py428 vision.UniformAugment(transforms=transforms_ua, num_ops=5)]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dvision.h1013 class UniformAugment final : public TensorTransform {
1018 explicit UniformAugment(const std::vector<TensorTransform *> &transforms, int32_t num_ops = 2);
1023 …explicit UniformAugment(const std::vector<std::shared_ptr<TensorTransform>> &transforms, int32_t n…
1028 …explicit UniformAugment(const std::vector<std::reference_wrapper<TensorTransform>> &transforms, in…
1031 ~UniformAugment() = default;
Dtransforms.h45 class UniformAugment; variable
59 friend class vision::UniformAugment;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dvision.cc1003 struct UniformAugment::Data {
1008 UniformAugment::UniformAugment(const std::vector<TensorTransform *> &transforms, int32_t num_ops) in UniformAugment() function in mindspore::dataset::vision::UniformAugment
1016 UniformAugment::UniformAugment(const std::vector<std::shared_ptr<TensorTransform>> &transforms, int… in UniformAugment() function in mindspore::dataset::vision::UniformAugment
1025 UniformAugment::UniformAugment(const std::vector<std::reference_wrapper<TensorTransform>> &transfor… in UniformAugment() function in mindspore::dataset::vision::UniformAugment
1032 std::shared_ptr<TensorOperation> UniformAugment::Parse() { in Parse()
/third_party/mindspore/mindspore/dataset/vision/
Dpy_transforms.py1663 class UniformAugment(py_transforms.PyTensorOperation): class
Dc_transforms.py1798 class UniformAugment(ImageTensorOperation): class