Home
last modified time | relevance | path

Searched refs:input_columns (Results 1 – 25 of 213) sorted by relevance

123456789

/third_party/mindspore/tests/ut/python/dataset/
Dtest_random_crop_and_resize.py67 data1 = data1.map(operations=decode_op, input_columns=["image"])
68 data1 = data1.map(operations=random_crop_and_resize_op, input_columns=["image"])
72 data2 = data2.map(operations=decode_op, input_columns=["image"])
106 data1 = data1.map(operations=transform1, input_columns=["image"])
115 data2 = data2.map(operations=transform2, input_columns=["image"])
148 data1 = data1.map(operations=transform1, input_columns=["image"])
166 data1 = data1.map(operations=decode_op, input_columns=["image"])
167 data1 = data1.map(operations=random_crop_and_resize_op, input_columns=["image"])
177 data2 = data2.map(operations=transform, input_columns=["image"])
202 data1 = data1.map(operations=decode_op, input_columns=["image"])
[all …]
Dtest_cutmix_batch_op.py54 data1 = data1.map(operations=hwc2chw_op, input_columns=["image"])
56 data1 = data1.map(operations=one_hot_op, input_columns=["label"])
59 data1 = data1.map(operations=cutmix_batch_op, input_columns=["image", "label"])
97 data1 = data1.map(operations=one_hot_op, input_columns=["label"])
99 data1 = data1.map(operations=rescale_op, input_columns=["image"])
102 data1 = data1.map(operations=cutmix_batch_op, input_columns=["image", "label"])
128 ds_original = ds_original.map(operations=[decode_op], input_columns=["image"])
130 ds_original = ds_original.map(operations=[resize_op], input_columns=["image"])
144 data1 = data1.map(operations=[decode_op], input_columns=["image"])
147 data1 = data1.map(operations=[resize_op], input_columns=["image"])
[all …]
Dtest_random_crop.py48 data1 = data1.map(operations=decode_op, input_columns=["image"])
49 data1 = data1.map(operations=random_crop_op, input_columns=["image"])
53 data2 = data2.map(operations=decode_op, input_columns=["image"])
79 data1 = data1.map(operations=transform1, input_columns=["image"])
88 data2 = data2.map(operations=transform2, input_columns=["image"])
114 data = data.map(operations=decode_op, input_columns=["image"])
115 data = data.map(operations=random_crop_op, input_columns=["image"])
141 data = data.map(operations=transform, input_columns=["image"])
163 data = data.map(operations=decode_op, input_columns=["image"])
164 data = data.map(operations=random_crop_op, input_columns=["image"])
[all …]
Dtest_config.py87 data1 = data1.map(operations=[c_vision.Decode(True)], input_columns=["image"])
92 data2 = data2.map(operations=[c_vision.Decode(True)], input_columns=["image"])
132 data1 = data1.map(operations=decode_op, input_columns=["image"])
133 data1 = data1.map(operations=random_crop_op, input_columns=["image"])
137 data2 = data2.map(operations=decode_op, input_columns=["image"])
139 data2 = data2.map(operations=random_crop_op, input_columns=["image"])
172 data1 = data1.map(operations=decode_op, input_columns=["image"])
173 data1 = data1.map(operations=random_crop_op, input_columns=["image"])
177 data2 = data2.map(operations=decode_op, input_columns=["image"])
181 data2 = data2.map(operations=random_crop_op2, input_columns=["image"])
[all …]
Dtest_random_horizontal_flip.py55 data1 = data1.map(operations=decode_op, input_columns=["image"])
56 data1 = data1.map(operations=random_horizontal_op, input_columns=["image"])
60 data2 = data2.map(operations=decode_op, input_columns=["image"])
94 data = data.map(operations=decode_op, input_columns=["image"])
95 data = data.map(operations=random_horizontal_op, input_columns=["image"])
121 data = data.map(operations=transform, input_columns=["image"])
143 data = data.map(operations=decode_op, input_columns=["image"])
144 data = data.map(operations=random_horizontal_op, input_columns=["image"])
167 data = data.map(operations=transform, input_columns=["image"])
183 data1 = data1.map(operations=decode_op, input_columns=["image"])
[all …]
Dtest_random_vertical_flip.py55 data1 = data1.map(operations=decode_op, input_columns=["image"])
56 data1 = data1.map(operations=random_vertical_op, input_columns=["image"])
60 data2 = data2.map(operations=decode_op, input_columns=["image"])
94 data = data.map(operations=decode_op, input_columns=["image"])
95 data = data.map(operations=random_horizontal_op, input_columns=["image"])
121 data = data.map(operations=transform, input_columns=["image"])
143 data = data.map(operations=decode_op, input_columns=["image"])
144 data = data.map(operations=random_horizontal_op, input_columns=["image"])
166 data = data.map(operations=transform, input_columns=["image"])
183 data1 = data1.map(operations=decode_op, input_columns=["image"])
[all …]
Dtest_project.py69 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
82 data1 = data1.map(operations=type_cast_op, input_columns=["col_sint64"])
97 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
98 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
99 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
100 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
104 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
105 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
106 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
107 data1 = data1.map(operations=type_cast_op, input_columns=["col_3d"])
[all …]
Dtest_autocontrast.py45 ds_original = data_set.map(operations=transforms_original, input_columns="image")
66 ds_auto_contrast = data_set.map(operations=transforms_auto_contrast, input_columns="image")
100 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
107 ds_auto_contrast_py = data_set.map(operations=transforms_op, input_columns="image")
120 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
122 ds_auto_contrast_c = data_set.map(operations=c_op, input_columns="image")
157 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
167 ds_auto_contrast_py = data_set.map(operations=transforms_op, input_columns="image")
181 input_columns=["image"])
183 ds_auto_contrast_c = data_set.map(operations=c_op, input_columns="image")
[all …]
Dtest_random_resize.py42 data1 = data1.map(operations=decode_op, input_columns=["image"])
44 data2 = data1.map(operations=resize_op, input_columns=["image"])
71 data = data.map(operations=decode_op, input_columns=["image"])
72 data = data.map(operations=resize_op, input_columns=["image"])
88 data = data.map(operations=ops.Duplicate(), input_columns=["image"],
93 data = data.map(operations=decode_op, input_columns=["image"])
94 data = data.map(operations=decode_op, input_columns=["image_copy"])
95 data = data.map(operations=resize_op, input_columns=["image", "image_copy"])
Dtest_center_crop.py43 data1 = data1.map(operations=decode_op, input_columns=["image"])
44 data1 = data1.map(operations=center_crop_op, input_columns=["image"])
48 data2 = data2.map(operations=decode_op, input_columns=["image"])
71 data1 = data1.map(operations=decode_op, input_columns=["image"])
72 data1 = data1.map(operations=center_crop_op, input_columns=["image"])
88 data1 = data1.map(operations=decode_op, input_columns=["image"])
89 data1 = data1.map(operations=center_crop_op, input_columns=["image"])
99 data2 = data2.map(operations=transform, input_columns=["image"])
131 data1 = data1.map(operations=transform, input_columns=["image"])
135 data1 = data1.map(operations=crop_gray, input_columns=["image"])
Dtest_mixup_op.py55 data1 = data1.map(operations=one_hot_op, input_columns=["label"])
58 data1 = data1.map(operations=mixup_batch_op, input_columns=["image", "label"])
85 ds_original = ds_original.map(operations=[decode_op], input_columns=["image"])
99 data1 = data1.map(operations=[decode_op], input_columns=["image"])
102 data1 = data1.map(operations=one_hot_op, input_columns=["label"])
106 data1 = data1.map(operations=mixup_batch_op, input_columns=["image", "label"])
146 data1 = data1.map(operations=one_hot_op, input_columns=["label"])
149 data1 = data1.map(operations=mixup_batch_op, input_columns=["image", "label"])
177 ds_original = ds_original.map(operations=[decode_op], input_columns=["image"])
191 data1 = data1.map(operations=[decode_op], input_columns=["image"])
[all …]
Dtest_to_number_op.py67 dataset = dataset.map(operations=text.ToNumber(ms_type), input_columns=["strings"])
83 dataset = dataset.map(operations=text.ToNumber(ms_type), input_columns=["strings"])
112 dataset = dataset.map(operations=text.ToNumber(ms_type), input_columns=["strings"])
117 dataset = dataset.map(operations=text.ToNumber(ms_type), input_columns=["strings"])
126 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[0]), input_columns=["strings"…
135 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[1]), input_columns=["strings"…
144 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[2]), input_columns=["strings"…
155 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[0]), input_columns=["strings"…
164 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[1]), input_columns=["strings"…
173 …dataset = dataset.map(operations=text.ToNumber(ms_non_integral_types[2]), input_columns=["strings"…
[all …]
Dtest_adjustgamma.py92 input_columns=["image"])
95 input_columns="image")
102 input_columns=["image"])
105 input_columns="image")
124 data_set = data_set.map(operations=[trans], input_columns=["image"])
136 data_set = data_set.map(operations=[trans], input_columns=["image"])
150 input_columns=["image"])
153 input_columns="image")
172 data_set = data_set.map(operations=[trans], input_columns=["image"])
190 ds1 = ds1.map(operations=transforms1, input_columns=["image"])
[all …]
Dtest_datasets_voc.py102 data2 = data2.map(operations=pyfunc1, input_columns=["image", "target"])
114 …data3 = data3.map(operations=pyfunc2, input_columns=["image", "target"], output_columns=["img1", "…
127 … data4 = data4.map(operations=pyfunc3, input_columns=["image", "target"], output_columns=["img1"],
141 data1 = data1.map(operations=resize_op, input_columns=["image"])
142 data1 = data1.map(operations=resize_op, input_columns=["target"])
159 data1 = data1.map(operations=resize_op, input_columns=["image"])
250 data = data.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
259 data = data.map(operations=vision.Decode(), input_columns=["image"], num_parallel_workers=1)
260 data = data.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
269 data = data.map(operations=exception_func, input_columns=["bbox"], num_parallel_workers=1)
[all …]
Dtest_random_rotation.py47 data1 = data1.map(operations=decode_op, input_columns=["image"])
48 data1 = data1.map(operations=random_rotation_op, input_columns=["image"])
52 data2 = data2.map(operations=decode_op, input_columns=["image"])
83 data1 = data1.map(operations=transform1, input_columns=["image"])
89 data2 = data2.map(operations=transform2, input_columns=["image"])
121 data1 = data1.map(operations=transform1, input_columns=["image"])
139 data1 = data1.map(operations=decode_op, input_columns=["image"])
140 data1 = data1.map(operations=random_rotation_op, input_columns=["image"])
165 data1 = data1.map(operations=decode_op, input_columns=["image"])
166 data1 = data1.map(operations=resize_op, input_columns=["image"])
[all …]
Dtest_equalize.py46 ds_original = data_set.map(operations=transforms_original, input_columns="image")
66 ds_equalize = data_set.map(operations=transforms_equalize, input_columns="image")
99 ds_original = data_set.map(operations=transforms_original, input_columns="image")
117 ds_equalize = data_set.map(operations=transform_equalize, input_columns="image")
146 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
148 ds_c_equalize = data_set.map(operations=C.Equalize(), input_columns="image")
162 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
169 ds_p_equalize = data_set.map(operations=transforms_p_equalize, input_columns="image")
202 … lambda img: np.array(img[:, :, 0])], input_columns=["image"])
204 data_set.map(operations=c_op, input_columns="image")
[all …]
Dtest_text_jieba_tokenizer.py55 data = data.map(operations=jieba_op, input_columns=["text"],
69 data = data.map(operations=jieba_op, input_columns=["text"],
82 data = data.map(operations=jieba_op, input_columns=["text"],
98 data = data.map(operations=jieba_op, input_columns=["text"],
112 data = data.map(operations=jieba_op, input_columns=["text"],
136 data = data.map(operations=jieba_op, input_columns=["text"],
154 data = data.map(operations=jieba_op, input_columns=["text"],
173 data = data.map(operations=jieba_op, input_columns=["text"],
189 data = data.map(operations=jieba_op, input_columns=["text"],
215 data = data.map(operations=jieba_op, input_columns=["text"],
[all …]
Dtest_datasets_coco.py252 data1 = data1.map(operations=resize_op, input_columns=["image"])
264 data1 = data1.map(operations=resize_op, input_columns=["image"])
335 … data1 = data1.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
344 … data1 = data1.map(operations=vision.Decode(), input_columns=["image"], num_parallel_workers=1)
345 … data1 = data1.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
354 data1 = data1.map(operations=exception_func, input_columns=["bbox"], num_parallel_workers=1)
363 …data1 = data1.map(operations=exception_func, input_columns=["category_id"], num_parallel_workers=1)
372 … data1 = data1.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
381 … data1 = data1.map(operations=vision.Decode(), input_columns=["image"], num_parallel_workers=1)
382 … data1 = data1.map(operations=exception_func, input_columns=["image"], num_parallel_workers=1)
[all …]
Dtest_pyfunc.py37 data1 = data1.map(operations=(lambda x: x + x), input_columns="col0", output_columns="out")
57 …data1 = data1.map(operations=(lambda x: (x, x + x)), input_columns=col, output_columns=["out0", "o…
81 data1 = data1.map(operations=(lambda x, y: x + y), input_columns=col, output_columns="out",
103 data1 = data1.map(operations=(lambda x, y: (x, x + y, x + y + 1)), input_columns=col,
129 data1 = data1.map(operations=(lambda x, y: (x, x + y, x + y + 1)), input_columns=col,
159 data1 = data1.map(operations=func_5, input_columns="col0", output_columns="out")
176 …data1 = data1.map(operations=[(lambda x: x + x), (lambda x: x + x)], input_columns="col0", output_…
198 data1 = data1.map(operations=(lambda x: x + x), input_columns="col0", output_columns="out",
224 data1 = data1.map(operations=(lambda x, y: (x, x + y, x + y + 1)), input_columns=col,
254 …ta1.map(operations=[(lambda x: x + x), (lambda x: x + 1), (lambda x: x + 2)], input_columns="col0",
[all …]
Dtest_random_sharpness.py46 ds_original = data.map(operations=transforms_original, input_columns="image")
70 ds_random_sharpness = data.map(operations=transforms_random_sharpness, input_columns="image")
111 data = data.map(operations=transform, input_columns=["image"])
135 ds_original = data.map(operations=transforms_original, input_columns="image")
158 ds_random_sharpness = data.map(operations=transforms_random_sharpness, input_columns="image")
197 data = data.map(operations=transforms, input_columns=["image"])
216 data = data.map(operations=[C.Decode(), C.Resize((200, 300))], input_columns=["image"])
225 ds_random_sharpness_py = data.map(operations=transforms_op, input_columns="image")
239 data = data.map(operations=[C.Decode(), C.Resize((200, 300))], input_columns=["image"])
241 ds_images_random_sharpness_c = data.map(operations=c_op, input_columns="image")
[all …]
Dtest_invert.py60 ds_original = data_set.map(operations=transforms_original, input_columns="image")
80 ds_invert = data_set.map(operations=transforms_invert, input_columns="image")
113 ds_original = data_set.map(operations=transforms_original, input_columns="image")
131 ds_invert = data_set.map(operations=transform_invert, input_columns="image")
160 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
162 ds_c_invert = data_set.map(operations=C.Invert(), input_columns="image")
176 data_set = data_set.map(operations=[C.Decode(), C.Resize((224, 224))], input_columns=["image"])
183 ds_p_invert = data_set.map(operations=transforms_p_invert, input_columns="image")
216 … lambda img: np.array(img[:, :, 0])], input_columns=["image"])
218 data_set.map(operations=c_op, input_columns="image")
[all …]
Dtest_mixup_label_smoothing.py42 dataset = dataset.map(operations=transform_label, input_columns=["label"])
72 ds1 = ds1.map(operations=decode_op, input_columns=["image"])
73 ds1 = ds1.map(operations=resize_op, input_columns=["image"])
74 ds1 = ds1.map(operations=one_hot_encode, input_columns=["label"])
84 ds1 = ds1.map(operations=transforms, input_columns=["image", "label"])
122 ds1 = ds1.map(operations=decode_op, input_columns=["image"])
123 ds1 = ds1.map(operations=resize_op, input_columns=["image"])
124 ds1 = ds1.map(operations=one_hot_encode, input_columns=["label"])
134 ds1 = ds1.map(operations=transforms, input_columns=["image", "label"])
Dtest_HWC2CHW.py66 data1 = data1.map(operations=decode_op, input_columns=["image"])
67 data1 = data1.map(operations=hwc2chw_op, input_columns=["image"])
71 data2 = data2.map(operations=decode_op, input_columns=["image"])
100 data1 = data1.map(operations=decode_op, input_columns=["image"])
101 data1 = data1.map(operations=hwc2chw_op, input_columns=["image"])
118 data1 = data1.map(operations=decode_op, input_columns=["image"])
119 data1 = data1.map(operations=hwc2chw_op, input_columns=["image"])
129 data2 = data2.map(operations=transform, input_columns=["image"])
Dtest_concatenate_op.py34 data = data.map(operations=concatenate_op, input_columns=["col"])
48 data = data.map(operations=concatenate_op, input_columns=["col"])
62 data = data.map(operations=concatenate_op, input_columns=["col"])
77 … data = data.map(operations=concatenate_op, input_columns=["col1", "col2"], column_order=["out1"],
92 … data = data.map(operations=concatenate_op, input_columns=["col1", "col2"], column_order=["out1"],
107 data = data.map(operations=concatenate_op, input_columns=["col"])
122 data = data.map(operations=concatenate_op, input_columns=["col"])
137 data = data.map(operations=concatenate_op, input_columns=["col"])
158 data = data.map(operations=concatenate_op, input_columns=["col"])
/third_party/mindspore/tests/st/networks/models/bert/src/
Ddataset.py44 data_set = data_set.map(operations=type_cast_op, input_columns="masked_lm_ids")
45 data_set = data_set.map(operations=type_cast_op, input_columns="masked_lm_positions")
46 data_set = data_set.map(operations=type_cast_op, input_columns="next_sentence_labels")
47 data_set = data_set.map(operations=type_cast_op, input_columns="segment_ids")
48 data_set = data_set.map(operations=type_cast_op, input_columns="input_mask")
49 data_set = data_set.map(operations=type_cast_op, input_columns="input_ids")

123456789