首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >是否可以将GradCam应用于TF Lite模型

是否可以将GradCam应用于TF Lite模型
EN

Stack Overflow用户
提问于 2021-04-26 22:21:47
回答 1查看 161关注 0票数 0

我一直在研究GradCam,我注意到大多数情况下都是在Keras/Tensorflow模型上使用的。但是,我有一个已编译为.tflite格式的tensorflow lite模型。我甚至不确定在编译后是否可以访问我的CNN层,因为我尝试使用keras库加载模型,但它只接受特定的文件类型,而不是确切的.tflite,因为它抛出了错误:

代码语言:javascript
复制
from tensorflow.keras.models import load_model
model = load_model("/content/drive/My Drive/tensorflow_lite_model.tflite")

它会给出错误:

代码语言:javascript
复制
OSError: SavedModel file does not exist

我尝试做的是使用model.summary打印.tflite模型,以确认我是否可以对模型层执行任何操作。如果是这样,那么我认为不可能将Grad-Cam与tensorflow lite模型一起使用。

因此,我想知道这是不是真的,或者我只是试图以错误的方式验证它?

EN

回答 1

Stack Overflow用户

发布于 2021-04-27 06:15:52

TFLite模型文件是与TensorFlow模型格式、keras和保存的模型不同的序列化格式。

由于您已经有了TFLite模型,因此需要使用TensorFlow Lite解释器API,而不是使用TensorFlow API。

代码语言:javascript
复制
interpreter = tf.lite.Interpreter(model_path="converted_model.tflite")
interpreter.allocate_tensors()

请参考此link了解详细信息。

TF GradCam模型可以转换为TFLite模型。从技术上讲,可以将任何TF模型转换为相应的TFLite模型。如果您在转换过程中有任何问题,请在tensorflow github上提交错误。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/67268414

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档