认识
2025年02月12日
一、认识
模型互转 基于 Python 训练好的模型，比如Tensorflow SavedModel、Tensorflow Frozen Model、Tensorflow Hub module .hub、Keras HDF5 .h5 等格式的模型, 可以通过 tensorflowjs_converter 转换为 JavaScript 可用模型 Layers Model 或者 Graph Model 的 .json 和 .bin, 然后可以在浏览器中运行。
二、准备
1. 下载 tensorflow/tensorflow 镜像
docker pull --platform tensorflow/tensorflow:2.10.0
2. 运行 tensorflow/tensorflow 容器
docker run -it --rm tensorflow/tensorflow:2.10.0 bash
3. 安装 tensorflow==2.0.0 tensorflowjs
pip install tensorflow==2.0.0 tensorflowjs
4. 将本地模型发送到 Docker 容器
docker cp /path/to/my_model.h5 <container_id>:/tmp/my_model.h5
三、转换
3.1 Keras HDF5 To JavaScript
.h5 转换为 tfjs_layers_model
tensorflowjs_converter --input_format=keras --output_format=tfjs_layers_model /tmp/my_model.h5 /tmp/layers_model
.h5 转换为 tfjs_graph_model
tensorflowjs_converter --input_format=keras --output_format=tfjs_graph_model /tmp/my_model.h5 /tmp/graph_model
tfjs_layers_model 转换为 .h5
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras /tmp/layers_model/model.json /tmp/my_model_new.h5