基于VGG网络对猫狗数据集分类

摘要

针对lenet网络对猫狗图像识别效果差的问题,本文构建了一种基于VGG网络目标检测模型的猫狗图像识别方法.通过采集猫狗图像,在PyTorch平台下,运用VGG网络提取特征.VGG网络参考了AlexNet、ZFNet、OverFeat等经典的网络后得出的.VGG的出现使小卷积核的应用正式走入大众的视野.本文中,我们将使用VGG-11进行猫狗数据集的分类.