/third_party/mindspore/tests/ut/python/dataset/ |
D | test_random_crop_and_resize.py | 67 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 …]
|
D | test_cutmix_batch_op.py | 54 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 …]
|
D | test_random_crop.py | 48 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 …]
|
D | test_config.py | 87 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 …]
|
D | test_random_horizontal_flip.py | 55 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 …]
|
D | test_random_vertical_flip.py | 55 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 …]
|
D | test_project.py | 69 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 …]
|
D | test_autocontrast.py | 45 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 …]
|
D | test_random_resize.py | 42 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"])
|
D | test_center_crop.py | 43 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"])
|
D | test_mixup_op.py | 55 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 …]
|
D | test_to_number_op.py | 67 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 …]
|
D | test_adjustgamma.py | 92 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 …]
|
D | test_datasets_voc.py | 102 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 …]
|
D | test_random_rotation.py | 47 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 …]
|
D | test_equalize.py | 46 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 …]
|
D | test_text_jieba_tokenizer.py | 55 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 …]
|
D | test_datasets_coco.py | 252 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 …]
|
D | test_pyfunc.py | 37 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 …]
|
D | test_random_sharpness.py | 46 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 …]
|
D | test_invert.py | 60 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 …]
|
D | test_mixup_label_smoothing.py | 42 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"])
|
D | test_HWC2CHW.py | 66 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"])
|
D | test_concatenate_op.py | 34 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/ |
D | dataset.py | 44 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")
|