如何将相同的扩充应用于两组相似的数据

时间:2019-07-19 19:17:50

标签: tensorflow-datasets

我正在尝试使用TensorFlow训练图像翻译模型,该数据集包含两个等长的图像列表,如何将它们都应用到地图上,以便它们获得相同的增强效果。

我试图创建一个变量来跟踪它是哪个迭代,以及使用种子随机数。

  

def load_and_preprocess_color(path):       图片= tf.read_file(路径)       图片= tf.image.decode_png(图片,通道= 3)       图片= tf.image.resize(图片,[224,224])       图片= tf.image.random_flip_left_right(图片)       图片/ = 255.0#归一化为[0,1]范围       返回图片

     

def load_and_preprocess_gray(path):       图片= tf.read_file(路径)       图片= tf.image.decode_png(图片,渠道= 1)       图片= tf.image.resize(图片,[224,224])       图片= tf.image.random_flip_left_right(图片)       图片/ = 255.0#归一化为[0,1]范围       返回图片

     

input_data_root = pathlib.Path('Images / Gray /')output_data_root =   pathlib.Path('Images / Color /')

     

input_image_paths = [input_data_root.iterdir()中项目的str(item)]   output_image_paths = [str(item)用于项目   output_data_root.iterdir()] image_count = len(input_image_paths)

     

input_path_ds = tf.data.Dataset.from_tensor_slices(input_image_paths)   output_path_ds =   tf.data.Dataset.from_tensor_slices(output_image_paths)

     

input_ds =   input_path_ds.map(load_and_preprocess_gray,num_parallel_calls = AUTOTUNE)   output_ds =   output_path_ds.map(load_and_preprocess_color,num_parallel_calls = AUTOTUNE`

我希望位于同一位置的两个数据要么翻转要么不翻转,但翻转状态与另一个相同

0 个答案:

没有答案