使用tensorflowjs_converter转换TF Hub的TFLite模型报错404问题咨询
错误原因
你遇到的404错误根本原因是你指定的两个TF Hub仓库存储的是TFLite格式模型,而tensorflowjs_converter使用--input_format=tf_hub参数时,默认会请求TF Hub接口返回原生SavedModel格式的模型包,这两个lite类模型没有对应的SavedModel下载端点,因此触发404。你测试的官方示例是标准SavedModel格式的TF Hub模型,因此可以正常拉取转换。
解决方法
- 第一步:手动下载对应TFLite模型到本地
直接打开模型页面点击下载按钮,将TFLite文件保存到本地,分别命名为stylization.tflite、cartoongan.tflite即可 - 第二步:升级tensorflowjs到最新版本,避免算子支持问题
执行命令:pip install --upgrade tensorflowjs - 第三步:使用tflite格式作为输入源执行转换
对应两个模型的转换命令分别为:tensorflowjs_converter --input_format=tflite ./stylization.tflite ./web_model/stylizationtensorflowjs_converter --input_format=tflite ./cartoongan.tflite ./web_model/cartoongan
注意事项
转换完成后的TensorFlow.js模型的输入输出张量维度、数值归一化规则和原TFLite模型完全一致,调用时直接沿用原TFLite模型的前处理、后处理逻辑即可。
内容的提问来源于stack exchange,提问作者Anran Xu
相关产品推荐
相关产品推荐

