Skip to content
/ gotalk Public
forked from liqichao/gotalk

Go+Docker实现的深度学习图片识别服务

Notifications You must be signed in to change notification settings

fendaq/gotalk

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

gotalk

这是我在 QCon 北京 2017 演讲 的配套代码。

基于 Go 实现了一个深度学习看图说话服务,即机器学习的 serving(inference) 部分,Tensorflow 的 Python 训练代码在 这里

代码了实现了 tensorflow 模型导入、输入输出和 LSTM beam search 功能,并实现了 web 服务。

我打包的 Docker 镜像中已经包含了所有模型文件,可以直接运行,见 docker hub

模型文件由 free_tf_model 工具 从 tf.train.Saver 保存的 checkpoint 生成,对于图说这个项目,训练代码保存的 checkpoint 无法直接用,需要通过 这个代码 中的 save_model.py 工具转化成 inference 模型后再调用 free_tf_model。

About

Go+Docker实现的深度学习图片识别服务

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Go 97.3%
  • Dockerfile 2.7%