Tensorflow tf.data基于另一个文件的内容在图像上进行操作

时间:2018-07-13 19:12:46

标签: python tensorflow deep-learning

我有3个文件img0_x.jpg img0_y.jpg img0_minmax.txt

img0_minmax.txt具有4行文本和4个值,如下所示:

xmin: -3.64646
xmax: 2.7953
ymin: -2.85746
ymax: 3.57254

我想拥有一个tf.data管道,该管道将加载两个jpg文件和txt文件,并基于xmin和xmax缩放_x.jpg的值,并基于ymin和ymax缩放_y.jpg的值。 / p>

我该怎么办? py_func是唯一的方法吗?

用于加载3个文件:

def _read_flows(x_img, y_img, minmax):
    x_img_str = tf.read_file(x_img)
    x_img_decoded = tf.image.decode_jpeg(x_img_str, channels=1)
    y_img_str = tf.read_file(y_img)
    y_img_decoded = tf.image.decode_jpeg(y_img_str, channels=1)
    return x_img_decoded, y_img_decoded, minmax

0 个答案:

没有答案