我在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
?
答案 0 :(得分:0)
我尝试过,我认为可以在内部运行tf.session,但是会减慢训练过程的速度。