查看: 296|回复: 1

[高级教程] 【TF】05 MNIST 模型的保存和加载

[复制链接]
MNIST 模型的保存和加载
本文获得作者CCFIVE授权转发
教程制作的这么详细,讲解通俗易懂,真是优质资源。
原文地址:http://learn.kittenbot.cn/zh_CN/latest/Tensorflow/05MnistLoad.html
感谢小喵科技,让小朋友也能感受机器学习。

上一节:【TF】经典案例:MNIST 手写数字识别
下一节:【TF】06 使用MobileNet模型进行物体识别


还记得我们上一节让大家保存的sb3项目吗?我们关了Kittenblock重现打开这个项目,发现识别的结果完全不对啊!怎么办呢?是数据丢了?

这是因为TensorFlow本质上是一个插件并不是Scratch3的一部分,sb3保存的时候也并没有将模型进行保存,所有东西都在你的显卡内存中如果重启了软件当然都丢失了喽。

那么这个时候我们只能重新点击绿旗帜建立模型,之后还有导入数据给TensorFlow引擎,最后开始重新训练。每次都要执行这个流程吗?这个效率很低啊~

其实Kittenblock有对应的方块将训练好的模型和进行保存。首先大家还是先耐心的将上一节的项目重新训练下:)

5.1. 保存训练的结果

我们可以使用如下的方块保存训练结果,点击这个方块后连续弹出两次保存框,一个是模型本身的拓扑结构,一个是模型的训练权重。

下面这两个文件就是模型的保存文件


5.2. 加载训练结果

我们重新建立一个工程,画笔那部分还是照着前一个项目那样拖一个代码

之后新建一个model变量,加入如下方块

截图201904131750254823.png

第一个参数是模型的拓扑结构文件的路径(.json),第二个参数就是模型的权重文件(.weights.bin)

我们打开sb3后只需要执行这一个方块就可以了,将你的机器学习项目分享给别的小伙伴的时候也记得带上这个模型的保存结果。

截图201904131750215459.png


gada888  版主

发表于 2019-4-29 22:06:54

厉害
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

为本项目制作心愿单
购买心愿单
心愿单 编辑
wifi气象站

硬件清单

btnicon
我也要做!
点击进入购买页面
上海智位机器人股份有限公司 沪ICP备09038501号-4

© 2013-2019 Comsenz Inc. Powered by Discuz! X3.4 Licensed

mail