这个项目基于pytorch框架,提供完整的VGG16图像分类代码,包括网络模型、训练和预测代码。附送鲜花数据集,已划分好训练、测试和验证数据,共包含4242张图像,分为洋甘菊、郁金香、玫瑰、向日葵和蒲公英五类。数据集来源于flicr、谷歌图像和yandex图像。图像分辨率约为320x240像素,不同比例,共800张照片每类。拿上代码即可运行。