OSError:追溯(最近一次呼叫最近)

时间:2020-10-26 14:32:28

标签: python tensorflow keras tensorflow2.0 tf.keras

我通过使用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 

它运行得很好,没有任何问题。突然开始出现此错误。我没有办法解决这个问题。我很高兴有人能给我一个解决错误的想法。

1 个答案:

答案 0 :(得分:0)

看起来还没有定义batch_size。尝试定义批处理大小,或使用#。

相关问题