Я портировал tfjs Google PoseNet на Python в праздничные дни.Демонстрационные приложения в хранилище автоматически загружают веса, замораживают график и сохраняют в файл модели.Вы можете взять эту модель и использовать ее в любом варианте TF.
Я написал Python-версию кода постобработки для нескольких человек, который использует векторизованные операции scipy / numpy для ускорения нескольких частей.Я не проводил исчерпывающего тестирования этой части, но с помощью ряда выборочных проверок различных тестовых изображений по сравнению с эталонным, и используя его для некоторых других источников, он кажется достаточно близким к оригиналу и быстрее:)
Python + TF на https://github.com/rwightman/posenet-python
Я также сделал преобразование PyTorch на https://github.com/rwightman/posenet-pytorch
И если вам в какой-то момент нужно искать порт CoreML, я начал сс кодом преобразования веса из этого проекта https://github.com/infocom-tpo/PoseNet-CoreML