我一直在研究GradCam,我注意到大多数情况下都是在Keras/Tensorflow模型上使用的。但是,我有一个已编译为.tflite格式的tensorflow lite模型。我甚至不确定在编译后是否可以访问我的CNN层,因为我尝试使用keras库加载模型,但它只接受特定的文件类型,而不是确切的.tflite,因为它抛出了错误:
from tensorflow.keras.models import load_model
model = load_model("/content/drive/My Drive/tensorflow_lite_model.tflite")它会给出错误:
OSError: SavedModel file does not exist我尝试做的是使用model.summary打印.tflite模型,以确认我是否可以对模型层执行任何操作。如果是这样,那么我认为不可能将Grad-Cam与tensorflow lite模型一起使用。
因此,我想知道这是不是真的,或者我只是试图以错误的方式验证它?
发布于 2021-04-27 06:15:52
TFLite模型文件是与TensorFlow模型格式、keras和保存的模型不同的序列化格式。
由于您已经有了TFLite模型,因此需要使用TensorFlow Lite解释器API,而不是使用TensorFlow API。
interpreter = tf.lite.Interpreter(model_path="converted_model.tflite")
interpreter.allocate_tensors()请参考此link了解详细信息。
TF GradCam模型可以转换为TFLite模型。从技术上讲,可以将任何TF模型转换为相应的TFLite模型。如果您在转换过程中有任何问题,请在tensorflow github上提交错误。
https://stackoverflow.com/questions/67268414
复制相似问题