深度学习tensorflow实战笔记（5）用预训练好的VGG-16模型提取图像特征

最新推荐文章于 2024-04-27 00:02:47 发布

蒲衣翁

最新推荐文章于 2024-04-27 00:02:47 发布

阅读量1.6w

点赞数 9

分类专栏： tensorflow-深度学习文章标签： VGG16 提取特征

本文链接：https://blog.csdn.net/gybheroin/article/details/79903044

版权

本文介绍了如何利用预训练的VGG-16模型从图像中提取特征。提供了模型结构的下载来源，并分享了网盘中的预训练模型文件。通过修改test_vgg16.py脚本，可以指定提取模型的特定层作为特征，简化了特征提取流程。

摘要由CSDN通过智能技术生成

上一篇博客介绍了如果使用自己训练好的模型用于图像分类和特征提取，但是有时候自己的数据集大小有限，所以更多的时候我们需要用VGG-16预训练好的模型提取特征，相关学者预训练好的模型使用的都是公开的标准数据集，所以我们直接用预训练的模型提取我们自己图像的特征，可以用于对图像进行描述。

1、首先就要下载模型结构

首先要做的就是下载训练好的模型结构和预训练好的模型，结构地址是：点击打开链接

预训练好的模型是VGG16.npy，这个文件需要单独下载，上面那个连接上有下载连接，不过一般打不开，我把网盘地址分享出来，希望可以帮助需要的人。

链接：https://pan.baidu.com/s/1rymDeCXiIGP-Uu7Qm3tWIQ 密码：sp1j

模型结构如下：

文件test_vgg16.py可以用于提取特征。其中vgg16.npy是需要单独下载的文件。

2、使用预训练的模型提取特征

打开test_vgg16.py，做如下修改：

import numpy as np
import tensorflow as tf

import vgg16
import utils

img1 = utils.load_image("./test_data/tiger.jpeg")
img2 = utils.

关注