跳到主要内容

认识

2025年02月12日
柏拉文
越努力，越幸运

一、认识​

在 TFJS 中，一个完整的模型通常包含:

  • model.json: 模型结构、权重路径等元数据(数据流图和权重清单文件)

  • group1-shard\*of\*: 二进制权重文件的集合

JavaScript 模型转换 主要是将 tfjs_layers_model 转换为 tfjs_graph_model。将 tfjs_layers_model 转换为 tfjs_graph_model 通常会加快浏览器和 Node.js 的推理速度，这要归功于生成 tfjs_graph_models 的图形优化。转换后的 tfjs_graph_model 不支持进一步训练。

二、准备​

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/layers_model/model.json

三、转换​

 tensorflowjs_converter \
    --input_format tfjs_layers_model \
    --output_format tfjs_graph_model \
    my_layers_model/model.json
    my_graph_model/