VGG-19 python代码 图片内容检测 预训练模型

【1】简介

  本文针对VGG-19网络，完成图像内容检测，对ImageNet的1000种常见物体进行识别。基本的思路是先下载预训练模型，使用已有的参数模型进行训练。

【2】预训练模型下载

  下载预训练模型vgg19.npy，这是后一步放入项目的文件夹下【Link】。

【3】代码与结果

  链接 （提取码：x1o8 ）
  注意可能会需要的问题：ValueError: Object arrays cannot be loaded when allow_pickle=False，以下提供两种解决方案

1. 升级Numpy；
2. 在np.load参数列表上加上allow_pickle=True，在上述网盘链接中下载的代码采用了这种方式；

  运行方法是在项目文件夹下使用cmd，执行以下命令：

python run.py -m folder -p testModel

  结果如下：