可以在运行fit_generator的同时在training_data_generator中运行“ With tf.Session as sess:”吗?

时间:2019-06-20 07:28:07

标签: python tensorflow keras

我在tensorboard中使用tf.Session,所以我无法启用Eager模式。

我需要使用tf.image.extract_image_patches提取大图像中的图像补丁。因此,在我的自定义训练生成器中,我添加了以下内容:

While True:
     num_patches = tf.image.extract_image_patches(input_big_pic, ksizes, strides, rates, patch_padding)
     With tf.Session as sess:
          inputs_after_tensor1 = sess.run(num_patches )
     .....some modifies for this ndarray "inputs_after_tensor1"....... 
     yield ({'input1': np.array(result_inputs_after_tensor1)})

我的损失函数正在减少,但是我在张量流上的输出图像没有改变,所以我想知道training_data_generator中的tf.Session是否会影响fit_generator

1 个答案:

答案 0 :(得分:0)

我尝试过,我认为可以在内部运行tf.session,但是会减慢训练过程的速度。