NumpyArrayIterartor 输入维度问题

Question

在网上一篇博客上跑了一下他的交通标志识别的训练代码，结果遇到这个问题，调试了好久没成功，求各位大佬指点~~万分感谢！

博客链接https://www.cnblogs.com/skyfsm/p/8051705.html#commentform

部分训练代码

    for imagePath in imagePaths:
        # load the image, pre-process it, and store it in the data list
        image = cv2.imread(imagePath)
        image = cv2.resize(image, (norm_size, norm_size))
        image = img_to_array(image)
        data.append(image)

H = model.fit_generator(aug.flow(trainX, trainY, batch_size=32),
        validation_data=(testX, testY), steps_per_epoch=len(trainX) // BS,
        epochs=EPOCHS, verbose=1)

if __name__=='__main__':
    args = args_parse()
    train_file_path = args["dataset_train"]
    test_file_path = args["dataset_test"]
    trainX,trainY = load_data(train_file_path)
    testX,testY = load_data(test_file_path)
    # construct the image generator for data augmentation
    aug = ImageDataGenerator(rotation_range=30, width_shift_range=0.1,
        height_shift_range=0.1, shear_range=0.2, zoom_range=0.2,
        horizontal_flip=True, fill_mode="nearest")
    train(aug,trainX,trainY,testX,testY,args)

Answer 1

https://blog.csdn.net/lqz790192593/article/details/89761801