摘要: 代码与解读: 先将所加载的图像进行一些预处理和变换。 使用预训练好的VGG16模型,将最后一层修改,使网络的结果可以输出两类,然后冻结最后一层前的层的参数,构成我们需要的网络模型。 创建损失函数和优化器,利用上述与处理好的train数据集和修改后的模型进行训练,然后用valid数据集测试训练好的模型 阅读全文
posted @ 2020-11-03 21:50 番茄蛋花汤 阅读(94) 评论(0) 推荐(0) 编辑