我有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