Home
last modified time | relevance | path

Searched refs:op_params (Results 1 – 25 of 121) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Daffine_ir.cc89 Status AffineOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *opera… in from_json() argument
90 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("degrees") != op_params.end(), "Failed to find degrees… in from_json()
91 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("translate") != op_params.end(), "Failed to find trans… in from_json()
92 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
93 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("shear") != op_params.end(), "Failed to find shear"); in from_json()
94 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("resample") != op_params.end(), "Failed to find resamp… in from_json()
95 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
96 float_t degrees = op_params["degrees"]; in from_json()
97 std::vector<float> translation = op_params["translate"]; in from_json()
98 float scale = op_params["scale"]; in from_json()
[all …]
Drandom_crop_decode_resize_ir.cc80 Status RandomCropDecodeResizeOperation::from_json(nlohmann::json op_params, in from_json() argument
82 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
83 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
84 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ratio") != op_params.end(), "Failed to find ratio"); in from_json()
85 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("interpolation") != op_params.end(), "Failed to find i… in from_json()
86 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("max_attempts") != op_params.end(), "Failed to find ma… in from_json()
87 std::vector<int32_t> size = op_params["size"]; in from_json()
88 std::vector<float> scale = op_params["scale"]; in from_json()
89 std::vector<float> ratio = op_params["ratio"]; in from_json()
90 InterpolationMode interpolation = static_cast<InterpolationMode>(op_params["interpolation"]); in from_json()
[all …]
Drotate_ir.cc105 Status RotateOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *opera… in from_json() argument
107 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("degree") != op_params.end(), "Failed to find degree"); in from_json()
108 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("resample") != op_params.end(), "Failed to find resamp… in from_json()
109 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("expand") != op_params.end(), "Failed to find expand"); in from_json()
110 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("center") != op_params.end(), "Failed to find center"); in from_json()
111 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
112 float degrees = op_params["degree"]; in from_json()
113 InterpolationMode resample = static_cast<InterpolationMode>(op_params["resample"]); in from_json()
114 bool expand = op_params["expand"]; in from_json()
115 std::vector<float> center = op_params["center"]; in from_json()
[all …]
Dslice_patches_ir.cc59 Status SlicePatchesOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
60 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("num_height") != op_params.end(), "Failed to find num_… in from_json()
61 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("num_width") != op_params.end(), "Failed to find num_w… in from_json()
62 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("slice_mode") != op_params.end(), "Failed to find slic… in from_json()
63 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
64 int32_t num_height = op_params["num_height"]; in from_json()
65 int32_t num_width = op_params["num_width"]; in from_json()
66 SliceMode slice_mode = static_cast<SliceMode>(op_params["slice_mode"]); in from_json()
67 uint8_t fill_value = op_params["fill_value"]; in from_json()
Drandom_resized_crop_with_bbox_ir.cc95 Status RandomResizedCropWithBBoxOperation::from_json(nlohmann::json op_params, in from_json() argument
97 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
98 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
99 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ratio") != op_params.end(), "Failed to find ratio"); in from_json()
100 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("interpolation") != op_params.end(), "Failed to find i… in from_json()
101 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("max_attempts") != op_params.end(), "Failed to find ma… in from_json()
102 std::vector<int32_t> size = op_params["size"]; in from_json()
103 std::vector<float> scale = op_params["scale"]; in from_json()
104 std::vector<float> ratio = op_params["ratio"]; in from_json()
105 InterpolationMode interpolation = static_cast<InterpolationMode>(op_params["interpolation"]); in from_json()
[all …]
Drandom_resized_crop_ir.cc100 Status RandomResizedCropOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperat… in from_json() argument
101 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
102 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
103 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ratio") != op_params.end(), "Failed to find ratio"); in from_json()
104 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("interpolation") != op_params.end(), "Failed to find i… in from_json()
105 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("max_attempts") != op_params.end(), "Failed to find ma… in from_json()
106 std::vector<int32_t> size = op_params["size"]; in from_json()
107 std::vector<float> scale = op_params["scale"]; in from_json()
108 std::vector<float> ratio = op_params["ratio"]; in from_json()
109 InterpolationMode interpolation = static_cast<InterpolationMode>(op_params["interpolation"]); in from_json()
[all …]
Drandom_crop_with_bbox_ir.cc129 Status RandomCropWithBBoxOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOpera… in from_json() argument
130 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
131 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding") != op_params.end(), "Failed to find padding… in from_json()
132 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("pad_if_needed") != op_params.end(), "Failed to find p… in from_json()
133 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
134 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding_mode") != op_params.end(), "Failed to find pa… in from_json()
135 std::vector<int32_t> size = op_params["size"]; in from_json()
136 std::vector<int32_t> padding = op_params["padding"]; in from_json()
137 bool pad_if_needed = op_params["pad_if_needed"]; in from_json()
138 std::vector<uint8_t> fill_value = op_params["fill_value"]; in from_json()
[all …]
Drandom_crop_ir.cc128 Status RandomCropOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *o… in from_json() argument
129 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
130 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding") != op_params.end(), "Failed to find padding… in from_json()
131 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("pad_if_needed") != op_params.end(), "Failed to find p… in from_json()
132 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
133 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding_mode") != op_params.end(), "Failed to find pa… in from_json()
134 std::vector<int32_t> size = op_params["size"]; in from_json()
135 std::vector<int32_t> padding = op_params["padding"]; in from_json()
136 bool pad_if_needed = op_params["pad_if_needed"]; in from_json()
137 std::vector<uint8_t> fill_value = op_params["fill_value"]; in from_json()
[all …]
Dsoftdvpp_decode_random_crop_resize_jpeg_ir.cc97 Status SoftDvppDecodeRandomCropResizeJpegOperation::from_json(nlohmann::json op_params, in from_json() argument
99 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
100 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
101 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ratio") != op_params.end(), "Failed to find ratio"); in from_json()
102 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("max_attempts") != op_params.end(), "Failed to find ma… in from_json()
103 std::vector<int32_t> size = op_params["size"]; in from_json()
104 std::vector<float> scale = op_params["scale"]; in from_json()
105 std::vector<float> ratio = op_params["ratio"]; in from_json()
106 int32_t max_attempts = op_params["max_attempts"]; in from_json()
Drandom_affine_ir.cc165 Status RandomAffineOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
166 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("degrees") != op_params.end(), "Failed to find degrees… in from_json()
167 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("translate") != op_params.end(), "Failed to find trans… in from_json()
168 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("scale") != op_params.end(), "Failed to find scale"); in from_json()
169 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("shear") != op_params.end(), "Failed to find shear"); in from_json()
170 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("resample") != op_params.end(), "Failed to find resamp… in from_json()
171 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
172 std::vector<float_t> degrees = op_params["degrees"]; in from_json()
173 std::vector<float_t> translate_range = op_params["translate"]; in from_json()
174 std::vector<float_t> scale_range = op_params["scale"]; in from_json()
[all …]
Drandom_rotation_ir.cc130 Status RandomRotationOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation… in from_json() argument
131 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("degrees") != op_params.end(), "Failed to find degrees… in from_json()
132 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("resample") != op_params.end(), "Failed to find resamp… in from_json()
133 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("expand") != op_params.end(), "Failed to find expand"); in from_json()
134 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("center") != op_params.end(), "Failed to find center"); in from_json()
135 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
136 std::vector<float> degrees = op_params["degrees"]; in from_json()
137 InterpolationMode resample = static_cast<InterpolationMode>(op_params["resample"]); in from_json()
138 bool expand = op_params["expand"]; in from_json()
139 std::vector<float> center = op_params["center"]; in from_json()
[all …]
Drandom_color_adjust_ir.cc97 Status RandomColorAdjustOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperat… in from_json() argument
98 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("brightness") != op_params.end(), "Failed to find brig… in from_json()
99 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("contrast") != op_params.end(), "Failed to find contra… in from_json()
100 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("saturation") != op_params.end(), "Failed to find satu… in from_json()
101 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("hue") != op_params.end(), "Failed to find hue"); in from_json()
102 std::vector<float> brightness = op_params["brightness"]; in from_json()
103 std::vector<float> contrast = op_params["contrast"]; in from_json()
104 std::vector<float> saturation = op_params["saturation"]; in from_json()
105 std::vector<float> hue = op_params["hue"]; in from_json()
Dresize_preserve_ar_ir.cc56 Status ResizePreserveAROperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperati… in from_json() argument
57 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("height") != op_params.end(), "Failed to find height"); in from_json()
58 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("width") != op_params.end(), "Failed to find width"); in from_json()
59 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("img_orientation") != op_params.end(), "Failed to find… in from_json()
60 int32_t height = op_params["height"]; in from_json()
61 int32_t width = op_params["width"]; in from_json()
62 int32_t img_orientation = op_params["img_orientation"]; in from_json()
Dcutmix_batch_ir.cc63 Status CutMixBatchOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *… in from_json() argument
64 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("image_batch_format") != op_params.end(), in from_json()
66 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("alpha") != op_params.end(), "Failed to find alpha"); in from_json()
67 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("prob") != op_params.end(), "Failed to find prob"); in from_json()
68 ImageBatchFormat image_batch = static_cast<ImageBatchFormat>(op_params["image_batch_format"]); in from_json()
69 float alpha = op_params["alpha"]; in from_json()
70 float prob = op_params["prob"]; in from_json()
Dnormalize_pad_ir.cc66 Status NormalizePadOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
67 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("mean") != op_params.end(), "Failed to find mean"); in from_json()
68 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("std") != op_params.end(), "Failed to find std"); in from_json()
69 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("dtype") != op_params.end(), "Failed to find dtype"); in from_json()
70 std::vector<float> mean = op_params["mean"]; in from_json()
71 std::vector<float> std = op_params["std"]; in from_json()
72 std::string dtype = op_params["dtype"]; in from_json()
Dascend_vision_ir.cc81 Status DvppCropJpegOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
82 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Fail to find size"); in from_json()
83 std::vector<uint32_t> resize = op_params["size"]; in from_json()
136 Status DvppDecodeResizeOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperati… in from_json() argument
137 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Fail to find size"); in from_json()
138 std::vector<uint32_t> resize = op_params["size"]; in from_json()
245 Status DvppDecodeResizeCropOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOpe… in from_json() argument
246 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("crop_size") != op_params.end(), "Fail to find crop_si… in from_json()
247 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("resize_size") != op_params.end(), "Fail to find resiz… in from_json()
248 std::vector<uint32_t> crop = op_params["crop_size"]; in from_json()
[all …]
Dadjust_gamma_ir.cc51 Status AdjustGammaOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *… in from_json() argument
52 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("gamma") != op_params.end(), "Failed to find gamma"); in from_json()
53 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("gain") != op_params.end(), "Failed to find gain"); in from_json()
54 float gamma = op_params["gamma"]; in from_json()
55 float gain = op_params["gain"]; in from_json()
Dnormalize_ir.cc48 Status NormalizeOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *op… in from_json() argument
49 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("mean") != op_params.end(), "Fail to find mean"); in from_json()
50 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("std") != op_params.end(), "Fail to find std"); in from_json()
51 std::vector<float> mean = op_params["mean"]; in from_json()
52 std::vector<float> std = op_params["std"]; in from_json()
Dpad_ir.cc108 Status PadOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operatio… in from_json() argument
109 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding") != op_params.end(), "Failed to find padding… in from_json()
110 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("fill_value") != op_params.end(), "Failed to find fill… in from_json()
111 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("padding_mode") != op_params.end(), "Failed to find pa… in from_json()
112 std::vector<int32_t> padding = op_params["padding"]; in from_json()
113 std::vector<uint8_t> fill_value = op_params["fill_value"]; in from_json()
114 BorderType padding_mode = static_cast<BorderType>(op_params["padding_mode"]); in from_json()
Dcutout_ir.cc55 Status CutOutOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *opera… in from_json() argument
56 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("length") != op_params.end(), "Failed to find length"); in from_json()
57 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("num_patches") != op_params.end(), "Failed to find num… in from_json()
58 int32_t length = op_params["length"]; in from_json()
59 int32_t num_patches = op_params["num_patches"]; in from_json()
Drescale_ir.cc59 Status RescaleOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *oper… in from_json() argument
60 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("rescale") != op_params.end(), "Failed to find rescale… in from_json()
61 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("shift") != op_params.end(), "Failed to find shift"); in from_json()
62 float rescale = op_params["rescale"]; in from_json()
63 float shift = op_params["shift"]; in from_json()
Dgaussian_blur_ir.cc65 Status GaussianBlurOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
66 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("kernel_size") != op_params.end(), "Failed to find ker… in from_json()
67 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("sigma") != op_params.end(), "Failed to find sigma"); in from_json()
68 std::vector<int32_t> kernel_size = op_params["kernel_size"]; in from_json()
69 std::vector<float> sigma = op_params["sigma"]; in from_json()
Dauto_contrast_ir.cc70 Status AutoContrastOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> … in from_json() argument
71 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("cutoff") != op_params.end(), "Failed to find cutoff"); in from_json()
72 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ignore") != op_params.end(), "Failed to find ignore"); in from_json()
73 float cutoff = op_params["cutoff"]; in from_json()
74 std::vector<uint32_t> ignore = op_params["ignore"]; in from_json()
Dbounding_box_augment_ir.cc61 Status BoundingBoxAugmentOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOpera… in from_json() argument
62 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("transform") != op_params.end(), "Failed to find trans… in from_json()
63 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("ratio") != op_params.end(), "Failed to find ratio"); in from_json()
66 json_operations.push_back(op_params["transform"]); in from_json()
68 float ratio = op_params["ratio"]; in from_json()
Dresize_ir.cc72 Status ResizeOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *opera… in from_json() argument
73 CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("size") != op_params.end(), "Failed to find size"); in from_json()
74 …CHECK_FAIL_RETURN_UNEXPECTED(op_params.find("interpolation") != op_params.end(), "Failed to find i… in from_json()
75 std::vector<int32_t> size = op_params["size"]; in from_json()
76 InterpolationMode interpolation = static_cast<InterpolationMode>(op_params["interpolation"]); in from_json()

12345