BodyPix是一种开放源代码的机器学习模型，该模型允许使用TensorFlow.js在浏览器中进行人和身体部位的分割。
我想将模型转换为.pb冻结图，以便在Python上使用它。
我该怎么做？
我尝试在不同的places上找到解决方案，但无法正常工作。
最佳答案
下载model.json文件
例如：https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/model-stride16.json
下载对应的权重
https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard1of23.bin
...
https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard23of23.bin
安装tfjs_graph_converter
来自https://github.com/patlevin/tfjs-to-tf
将模型转换为.pb文件
tfjs_graph_converter path / to / js / model path / to / frozen / model.pb