Tensorflow-VGG16-模型py文件调用

1、pb模型生成

用tensorflow训练好模型,并生成pb文件。
**注意:**tensorflow训练好的模型保存方式有两种:ckpt格式和pb格式,其中前者主要用于暂存我们训练的临时数据,避免发生意外导致训练终止,前面的努力全部白费掉了。而后者常用于将模型固化,提供离线预测,用户只要提供一个输入,通过模型就可以得到一个预测结果。

**Step1:**运行datagenerator.py,将图片转化成tfrecorder格式,生成两个文件,分别是train.tfrecorder和val.tfrecorder,分别是验证集和测试集,标签也包含在里面了。
在这里插入图片描述
**Setp2:**运行vgg16.py进行网络训练,层数可以自行调节,模型中的参数、图片大小可以根据需要来调节,注意要在训练中分别给输入、输出起一个名字,方便后面的测试。训练完成后会生成一个.pb文件。
在这里插入图片描述
**Step3:**运行validate_image.py测试模型效果。

2、调用

**Step1:**在VS中新建一个控制台程序或MFC程序,由于训练是在win64下进行的,因此环境配置也要在win64下,具体操作如下:
1.1加载库
在这里插入图片描述
1.2头文件
在这里插入图片描述
1.3附加依赖项
打开Anaconda安装路径“C:\Users\Administrator\Anaconda3\libs”会发现其中只有python35.lib,需要将其复制,然后重命名为python35_d.lib。
然后再项目——属性——连接器——输入中,将其添加到附加依赖项中。
在这里插入图片描述
1.4新建.cpp文件与.py文件
注意py文件名需要和cpp中对应,并将.pb文件以及.py文件放到项目文件夹的Debug文件夹下,例如:
在这里插入图片描述
运行程序,报错,进行以下检查:
1)检查.py文件有没有格式错误
2)检查代码中的文件路径是否正确
3)要求调用模型的电脑环境安装有tensorflow
4)检查环境配置
最终输出
在这里插入图片描述
完成!

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值