【Deep Learning笔记】用Inception-V3模型进行图像分类

文章目录

  • 1 Inception-V3模型简介
  • 2 ImageNet数据集简介
  • 3 工程文件
  • 4 运行
  • 5 结果

1 Inception-V3模型简介

本例使用预训练好的深度神经网络Inception-v3模型来进行图像分类。Inception-v3模型在一台配有 8 Tesla K40 GPUs,大概价值$30,000的野兽级计算机上训练了几个星期,因此不可能在一台普通的PC上训练。我们将会下载预训练好的Inception模型,然后用它来做图像分类。

Inception-v3模型大约有2500万个参数,分类一张图像就用了50亿的乘加指令。在一台没有GPU的现代PC上,分类一张图像转眼就能完成。

2 ImageNet数据集简介

ImageNet数据集包含1500万张图片,22000个类别。其子集对应的是目前最权威的图片分类竞赛LSVRC,包含100万张图片和1000个类别。

谷歌在大型图像数据库ImageNet上训练好了一个Inception-v3模型,这个模型可以直接用来进行图像分类。

输入:一张图片;

输出:与输入图相匹配的库内图片名及匹配分数。

3 工程文件

【Deep Learning笔记】用Inception-V3模型进行图像分类_第1张图片
在这里插入图片描述
文件:就是训练好的Inception-v3模型;
在这里插入图片描述
文件:为类别文件,共包含2万多种类别,内容如下:
【Deep Learning笔记】用Inception-V3模型进行图像分类_第2张图片
在这里插入图片描述
为进行分类的代码;

在这里插入图片描述
为分类的输入图;

在这里插入图片描述
为镜像文件;
在这里插入图片描述
【Deep Learning笔记】用Inception-V3模型进行图像分类_第3张图片

4 运行

在这里插入图片描述
【Deep Learning笔记】用Inception-V3模型进行图像分类_第4张图片

5 结果

【Deep Learning笔记】用Inception-V3模型进行图像分类_第5张图片
运行,获得库内分数最高的前5个(即获得输入图的类别),结果如下:

在这里插入图片描述
修改目标图为熊猫

  1. 如果将目标图改为熊猫,例图为:
    在这里插入图片描述
    打开Classify.py文件,代码做如下修改:
    【Deep Learning笔记】用Inception-V3模型进行图像分类_第6张图片
    运行结果如下:

【Deep Learning笔记】用Inception-V3模型进行图像分类_第7张图片
2. 在 文件内,查找panda,即可找到所有panda:
【Deep Learning笔记】用Inception-V3模型进行图像分类_第8张图片

你可能感兴趣的:(Deep,learning)