tf.image.crop_and_resize导致提前停止

时间:2018-10-17 16:47:01

标签: tensorflow

我在张量流中使用估计器框架。我正在输入图像数据集

images.shuffle(buffer_size=3000).batch(100).repeat()

进入火车功能

estimator.train(input_fn=train_input_fn, steps=2000)

因此,我的批次大小为100,并且希望执行2000步训练。一切都会按预期进行,但是只要我添加此代码块

images = tf.image.crop_and_resize(
    params["images"],
    boxes = np.random.rand(100, 4),
    box_ind = [x for x in range(100)],
    crop_size = (model_height, model_width)
)

对于我的模型功能,训练在所有图像处理一次后停止,而不是在2000个步骤之后停止。我不明白为什么?

0 个答案:

没有答案
相关问题