这是我使用resnet50.h5进行训练新的模型时出现的问题,而出现此问题的原因是模型的resnet的类别是1000类。而此时的类别是7类,类别不匹配造成的,所以解决方法是:在加载模型权重是设置by_name=True,意思是允许进行fine-tuning微调
model.load_weights(weight_path,by_name=True)
ValueError: decode_predictions:a 2D array of shape (samples,, 1000)). Found array with shape: (1, 7)
最新推荐文章于 2024-08-18 16:28:24 发布