使用Keras预训练好的模型进行目标类别预测详解

更新时间：2020年06月27日 09:24:41 作者：kongfd76

这篇文章主要介绍了使用Keras预训练好的模型进行目标类别预测详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

前言

最近开始学习深度学习相关的内容，各种书籍、教程下来到目前也有了一些基本的理解。参考Keras的官方文档自己做一个使用application的小例子，能够对图片进行识别，并给出可能性最大的分类。

闲言少叙，开始写代码

环境搭建相关就此省去，网上非常多。我觉得没啥难度

from keras.applications.resnet50 import ResNet50
from keras.preprocessing import image
from keras.applications.resnet50 import preprocess_input, decode_predictions
import numpy as np

导入权重，首次会从网络进行下载，不过速度还是挺快的，使用ImageNet的数据集

model = ResNet50(weights='imagenet')

定义一个函数读取图片文件并处理。这里需要安装PLI的库。 pip install Pillow ,不然会报错

def load_image(img_path):
  img = image.load_img(img_path, target_size=(224, 224))
  x = image.img_to_array(img)
  x = np.expand_dims(x, axis=0)
  x = preprocess_input(x)
  return x

加载一个图片文件，默认在当前路径寻找

x=load_image('zebra.jpg')

哈哈，开始预测了！激动人心啊

preds = model.predict(x)

执行速度很快，现在看看结果

print('Predicted:', decode_predictions(preds, top=3)[0])

Predicted: [(‘n02391049', ‘zebra', 0.99566585), (‘n02423022', ‘gazelle', 0.0010297714), (‘n01518878', ‘ostrich', 0.00067320856)]

准确率还是不错，后续还测试了一些飞机之类的图片，总体来讲马马虎虎！

是不是非常简单，确实很简单！

补充知识：模型训练loss先迅速下降后一直上升

loss函数走势如下：

检查代码没什么问题，分析应该是陷入了局部最优，把学习率调低一点就好了，从0.01调到了0.001

以上这篇使用Keras预训练好的模型进行目标类别预测详解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python各种excel写入方式的速度对比
这篇文章主要介绍了python各种excel写入方式的速度对比，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-11-11
图文详解matlab原始处理图像几何变换
Matlab 擅长于操作矩阵,而图像其实就是矩阵,这篇文章主要给大家介绍了关于matlab原始处理图像几何变换的相关资料,需要的朋友可以参考下
2021-07-07
一文带你掌握Python中文词频统计
词频统计是指在文本中计算每个词出现的次数。这篇文章主要带大家了解一下Python实现中文词频统计的方法，感兴趣的小伙伴可以了解一下
2023-02-02
python执行外部程序的常用方法小结
这篇文章主要介绍了python执行外部程序的常用方法,实例总结了Python使用os模块及commands模块执行外部程序的技巧,非常具有实用价值,需要的朋友可以参考下
2015-03-03
python模块shutil函数应用示例详解教程
这篇文章主要为大家介绍了python模块中shutil函数的应用示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2021-10-10
解决python logging遇到的坑日志重复打印问题
这篇文章主要介绍了解决python logging遇到的坑日志重复打印问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
Python中Django 后台自定义表单控件
本篇文章主要介绍了Python中Django 后台自定义表单控件，其实 django 已经为我们提供了一些可用的表单控件，比如：多选框、单选按钮等，有兴趣的开业了解一下。
2017-03-03
利用python实现3种梯度下降算法
梯度下降法是一种优化算法,用于求解函数的最小值或最大值,它通过迭代的方式,沿着函数的梯度方向逐步调整参数,以找到函数的极值点,本文给大家介绍了利用python实现3种梯度下降算法,需要的朋友可以参考下
2023-12-12
13个你可能未使用过的Python特性分享
Python 是顶级编程语言之一，它具有许多程序员从未使用过的许多隐藏功能。在这篇博客中，将和大家分享一下你可能从未使用过的13 个 Python 特性，感兴趣的可以学习一下
2022-11-11
python列出目录下指定文件与子目录的方法
这篇文章主要介绍了python列出目录下指定文件与子目录的方法,涉及Python使用os模块与glob操作目录与文件的技巧,需要的朋友可以参考下
2015-07-07