使用保存的一种模式的权重来训练新的深度模型

时间:2021-03-29 19:02:49

标签: python tensorflow keras deep-learning computer-vision

我是 tensorflow 的新手。我试图通过将 VGG16 模型训练到包含 9 个类的 tensorflow 数据集上可用的 deepweeds 数据集来使用 VGG16 模型进行图像分类。训练模型后,我将其权重保存在 .h5 扩展名中。现在,我尝试使用保存的权重作为默认权重,在包含三个类的新数据集上训练 vgg16 模型。 Here is the sample code that I am using to train on new dataset

But everytime, I get some errors such as:

感谢您的帮助和建议

0 个答案:

没有答案
相关问题