tensorflow - BodyPix:实时人物分割

Question

BodyPix是一种开源机器学习模型，它允许使用 TensorFlow.js 在浏览器中对人和 body 部位进行分割。
我想将模型转换为 .pb 卡住图，以便在 Python 上使用它。

我该怎么做？

我尝试在不同的 places 上找到解决方案，但不工作。

Answer 1

下载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