Python 3.x tf.image.decode_image的替代方案？

我发现在tensorflow 1.14版本下，

tf.image.decode\u image

的运行时间非常长。但是在2.0版本下是可以的

import os
def create_dataset(csv_filename):
    # Do things here, return the dataset
    # return dataset
    csvdataset = tf.data.experimental.CsvDataset(csv_filename,[tf.string, tf.string])
    list_raw = []
    list_reg = []
    for element in csvdataset:
        raw_element = tf.io.decode_image(tf.io.read_file(root_dir+tf.strings.substr(element[0], 2, tf.size(element[0])-2)), dtype=tf.dtypes.uint16)       
       # seg_element = tf.image.decode_image(tf.io.read_file(root_dir+tf.strings.substr(element[1], 2, tf.size(element[0])-2)), dtype=tf.dtypes.uint8)
       # list_raw.append(raw_element)
       # list_reg.append(seg_element)
    #dataset = tf.data.Dataset.from_tensor_slices((list_raw, list_reg))
    #return dataset