javascript模型转换前需要安装conda,conda下载网址Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
conda基本使用
conda create -n tfjs python=3.6.8 //创建虚拟环境
conda info --envs //查看环境
conda activate tfjs //进入环境
conda deactivate //退出环境
conda remove -n tfjs --all //删除环境
模型转换
conda activate tfjs //进入环境
pip install tensorflowjs //在环境中安装tensorflowjs
//python模型转换为javascript模型
//设置输入的模型格式 设置输出模型格式 输入模型路径 输出模型路径
tensorflowjs_converter --input_format=keras --output_format=tfjs_layers_model .\keras.h5 ./web_model2
tensorflowjs_converter --input_format=keras --output_format=tfjs_graph_model .\keras.h5 ./web_graph_model2
//javascript模型转为python模型
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=keras ./web_model2/model.json .\keras2.h5
//javascript模型分片
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=tfjs_layers_model --weight_shard_size_bytes=100000 ./web_model/model.json ./sharded_model
//javascript量化模型,缩小模型体积,精度会受到损失
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=tfjs_layers_model --quantization_bytes=2 ./web_model/model.json ./quantization_model
//javascript模型转换为graph模型,速度更快
tensorflowjs_converter --input_format=tfjs_layers_model --output_format=tfjs_graph_model ./web_model/model.json ./graph_model