您当前的位置: 首页 >  tensorflow

耐心的小黑

暂无认证

  • 3浏览

    0关注

    323博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

tensorflow2.x踩坑记录二:加载含Lambda层的模型时,出现name tf is not defined

耐心的小黑 发布时间:2020-08-26 10:06:13 ,浏览量:3

一、问题描述:

最近在学习一个TF2.x版本的yolov4项目,当我看明白主程序的整个检测过程以及一些网络搭建的代码之后,就准备将模型保存成.pb或者.h5格式,然后在其他程序中直接调用进行推理。保存的时候没有出现问题,成功的保存成了.pb文件,但是在使用tf.keras.models.load_model加载模型的时候,出现了name tf is not defined的报错。

保存、加载以及推理的大概流程如下:

self.yolo_model.save('./saved_model')
model=tf.keras.models.load_model('./saved_model'
关注
打赏
1640088279
查看更多评论
立即登录/注册

微信扫码登录

0.1292s