ValueError: decode_predictions:a 2D array of shape (samples,, 1000)). Found array with shape: (1, 7)

这是我使用resnet50.h5进行训练新的模型时出现的问题，而出现此问题的原因是模型的resnet的类别是1000类。而此时的类别是7类，类别不匹配造成的，所以解决方法是：在加载模型权重是设置by_name=True，意思是允许进行fine-tuning微调
model.load_weights(weight_path,by_name=True)

版权声明：本文为ShaoDu原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。