【发布时间】:2021-02-08 19:55:14
【问题描述】:
我使用 Keras 和 Tensorflow 开发的 CNN 的数据增强功能将这个错误作为 UnknownError 得到,如下所述。
testGenerator = ImageDataGenerator(
rescale = 1./255.0,
rotation_range = 45,
horizontal_flip = True,
vertical_flip = True,
zoom_range = [0.5,1.0],
data_format = 'channels_last',
dtype = tf.float32,
)
test_generator = dataGenerator.flow_from_directory(
"D:\\Data\\", target_size = (image_height, image_width), batch_size=batch_size,
class_mode = 'sparse', shuffle=True, seed=123, color_mode = 'grayscale',
)
print("No of augmented training data: " + str(len(test_generator)))
执行在model.evaluation函数之后终止。
UnknownError Traceback (most recent call last)
<ipython-input-9-455b71844aa2> in <module>
12 evaluation = model.evaluate(
13 test_generator,
---> 14 batch_size=batch_size,
15 )
16
它早些时候运行良好,没有任何问题。突然开始出现这个错误。我不知道解决这个问题。如果有人可以给我一个修复错误的想法,我很高兴。
【问题讨论】:
-
也许删除 model.evaluate 中的 batch_size 参数?生成器具有无法更改的固定批量大小
-
@Dr.Snoopy 我也试过了。然后它将错误指出到上一行 test_generator。我认为我的 test_generator 出了点问题。我也删除了里面的batch_size。但仍然有错误。
-
然后包括完整的回溯和重现问题的代码
标签: python tensorflow keras tensorflow2.0 tf.keras