使用Keras打造一个实时可用交通标志识别App

向AI转型的程序员都关注了这个号????????????

机器学习AI算法工程   公众号:datayx

 

 

大纲

数据来源

数据来源

 

对于我们做应用的人而言,最重要的应该就是数据。数据往往是一个算法公司的主要财产之一。那么如何为自己的问题获取对应的数据呢?先说结论:大型公开数据集 > 迁移学习 > 自己标注。

 

  1. Google最近推出的数据集搜索引擎

  2. Kaggle 上面的一些比赛也会有一些公开数据

  3. Google检索

     

  4. 实在没有办法可以自己标注数据,但是这个是成本很大的问题,还有准确度的问题。当然这只是针对小公司而言,对于大公司数据也是壁垒之一。

 

使用Keras打造一个实时可用交通标志识别App

 

注意这里我们只将这个数据用来学习使用。

 

项目代码 获取方式:

关注微信公众号 datayx  然后回复  交通标志  即可获取。

AI项目体验地址 https://loveai.tech

 

 

模型训练

这里我们使用Keras框架,Tensorflow作为后端来进行训练,其实对于一般做移动端应用的公司我觉得使用Keras,然后转换到移动端推理框架挺方便的。

 

 

 

使用Keras打造一个实时可用交通标志识别App

 

 

我的网络结构:simplenet.py

 

使用Keras打造一个实时可用交通标志识别App

 

 

准备好数据和网络配置文件之后在tran.py训练脚本中传入相应的参数,直接训练便可。

训练100 epochs之后就有0.945-0.95的准确度了,说明我们的模型效果还可以。

 

训练好模型之后一般需要在真实环境测试一下:测试脚本:

 

使用Keras打造一个实时可用交通标志识别App

 

使用Keras打造一个实时可用交通标志识别App

 

 

使用Keras打造一个实时可用交通标志识别App

 

 

看起来还可以哈

 

一般很多人的文章调完参数,达到一定的准确度,观察一些测试数据,就不介绍了。然而你有这个模型,如何将它应用到生产环境中还有一段路要走。接下来的部分就介绍如何将训练好的模型移植到移动端,打造一个真正实时可用的App。

 

模型转换

这一小结介绍如何将模型转换到移动端可用框架。

 

我们遇到的第一个问题是需要将Keras模型转换到CoreML可用的格式, 这里提供一个转换脚本(版本不同会有接口的变换, 这里是python2, Keras 2.1.6, tensorflow 1.12.0):

 

使用Keras打造一个实时可用交通标志识别App

 

正确转换之后我们就得到CoreML下可用的深度学习模型了,剩下的只需要在IOS工程中正确调用便可,稍微有些IOS 开发相关的知识就能完成。这里只是抛砖引玉,要实现其他功能的应用,流程也大致如此。

原文地址 https://www.jianshu.com/p/f5bcb974831e

 

 

 


 

相关文章:

  • 2021-10-14
  • 2022-12-23
  • 2021-12-31
  • 2021-12-27
  • 2021-09-25
  • 2021-08-05
  • 2021-12-04
  • 2021-05-30
猜你喜欢
  • 2021-12-05
  • 2021-11-14
  • 2021-12-13
  • 2021-11-23
  • 2021-07-13
  • 2021-07-17
  • 2022-02-09
相关资源
相似解决方案