马克萨/百香果图像分类的改进研究
1. 模型过拟合问题
在模型训练过程中,出现了过拟合现象。模型能够完美地对训练图像进行分类,但在验证集上的准确率仅为 0.65。这表明模型在训练数据上表现良好,但缺乏泛化能力,无法很好地适应新的数据。
2. VGG16 模型实验
- 模型架构 :在实验中,我们在 VGG16 的扁平化层之后添加了 1 个密集层,并在输出层之前添加了 1 个丢弃层,同时冻结了 VGG16 的基础卷积层。
- 参数调优 :我们对不同的参数组合进行了实验,共训练了 64 个模型。以下是不同参数对模型性能的影响:
- 优化器 :选择了 Adam 和 SGD 两种优化器,在相同参数下进行比较。结果显示,Adam 优化器的性能优于 SGD,Adam 优化器的准确率达到 0.97,而 SGD 仅为 0.75。具体参数如下表所示:
| 优化器 | 准确率 | 密集层神经元数量 | 丢弃率 | 轮数 | 学习率 | 批次大小 |
| ---- | ---- | ---- | ---- | ---- | ---- | ---- |
| Adam | 0.97 | 512 | 0.2 | 20 | 0.01 | 100 |
| SGD | 0.75 | 512 | 0.2 | 20 | 0.01 | 100 | - 密集层神经元数量 :尝试了 512 和 1024 两个不同的值。结果表明,较低数量的神经元(512
- 优化器 :选择了 Adam 和 SGD 两种优化器,在相同参数下进行比较。结果显示,Adam 优化器的性能优于 SGD,Adam 优化器的准确率达到 0.97,而 SGD 仅为 0.75。具体参数如下表所示:
超级会员免费看
订阅专栏 解锁全文
1097

被折叠的 条评论
为什么被折叠?



