tensorflow模型转tfjs文件

tensorflow模型转tfjs文件

安装tensorflowjs

pip安装tensorflowjs，打开终端：

# 安装库
pip install tensorflowjs

转换模型

终端不要退出，直接通过tensorflowjs_converter进行转换

tensorflowjs_converter --input_format=tf_saved_model --output_node_names='detection_boxes,detection_classes,detection_features,detection_multiclass_scores,detection_scores,' --saved_model_tags=serve --output_format=tfjs_graph_model path_to_your_exported_inference_model_dir/saved_model path_to_save_your_tfjs_model

参数解释：

1. input_format：就是输入模型的格式，我的tensorflow是用save_model保存的就是tf_saved_model
2. output_node_names：要输出的节点。
   我填了，不填也可以生成tfjs模型，但是后面使用可能会有什么报错。不知道填什么的话，我是用这个来看输出节点的。

https://blog.csdn.net/qq_41158484/article/details/126728409?spm=1001.2014.3001.5502

3. output_format：输出的模型格式，tfjs_graph_model
4. saved_model_tags：默认为serve，用来区别不同的MetaGraphDef
5. 后面两个参数分别是tensorflow模型的地址 和 要输出的tfjs模型的地址：
   tensorflow模型的地址指向 saved_model 文件夹
   tfjs模型指向生成的模型文件位置

参考

https://blog.csdn.net/qq_35496811/article/details/122626934