def test_random_resize_op(plot=False): """ Test random_resize_op """ logger.info("Test resize") data1 = ds.TFRecordDataset(DATA_DIR, SCHEMA_DIR, columns_list=["image"], shuffle=False) # define map operations decode_op = vision.Decode() resize_op = vision.RandomResize(10) # apply map operations on images data1 = data1.map(input_columns=["image"], operations=decode_op) data2 = data1.map(input_columns=["image"], operations=resize_op) image_original = [] image_resized = [] num_iter = 0 for item1, item2 in zip(data1.create_dict_iterator(), data2.create_dict_iterator()): image_1 = item1["image"] image_2 = item2["image"] image_original.append(image_1) image_resized.append(image_2) num_iter += 1 if plot: visualize_list(image_original, image_resized)
def test_random_resize_op(): """ Test random_resize_op """ logger.info("Test resize") data1 = ds.TFRecordDataset(DATA_DIR, SCHEMA_DIR, columns_list=["image"], shuffle=False) # define map operations decode_op = vision.Decode() resize_op = vision.RandomResize(10) # apply map operations on images data1 = data1.map(input_columns=["image"], operations=decode_op) data1 = data1.map(input_columns=["image"], operations=resize_op) num_iter = 0 for item in data1.create_dict_iterator(): _ = item["image"] # Uncomment below line if you want to visualize images # visualize(image_de_resized, image_np_resized, mse) num_iter += 1
def test_random_resize_md5(): """ Test RandomResize with md5 check """ logger.info("Test RandomResize with md5 check") original_seed = config_get_set_seed(5) original_num_parallel_workers = config_get_set_num_parallel_workers(1) # Generate dataset data = ds.TFRecordDataset(DATA_DIR, SCHEMA_DIR, columns_list=["image"], shuffle=False) decode_op = vision.Decode() resize_op = vision.RandomResize(10) data = data.map(input_columns=["image"], operations=decode_op) data = data.map(input_columns=["image"], operations=resize_op) # Compare with expected md5 from images filename = "random_resize_01_result.npz" save_and_check_md5(data, filename, generate_golden=GENERATE_GOLDEN) # Restore configuration ds.config.set_seed(original_seed) ds.config.set_num_parallel_workers(original_num_parallel_workers)