基于改进VGG16图像分类方法研究

摘要

针对神经网络模型在训练过程中遇到的收敛速度慢和测试样本不平衡导致的准确率降低问题，提出了一种基于改进VGG16图像分类模型的LBF-VGG16(Leaky-Bactch-Focal-VGG16)。该模型将原Relu激活函数替换为Leaky Relu,并在卷积层与激活函数之间引入BN层，以优化收敛效果。在训练过程中，采用SGD优化器，并融入Focal Loss损失函数。试验结果表明，LBF-VGG16模型在分类效果和收敛速度方面较改进前均有显著提升

关键词

图像分类 / 计算机视觉 / VGG16 / Leaky Relu / Focal Loss