Skip to content

Customize posetrackinggpu example to use heavy or light model #4618

Open
@mruvim

Description

@mruvim

OS Platform and Distribution

iOS 15

Compiler version

No response

Programming Language and version

ObjC/Swift

Installed using virtualenv? pip? Conda?(if python)

No response

MediaPipe version

0.10.2

Bazel version

6.2.1

XCode and Tulsi versions (if iOS)

14

Android SDK and NDK versions (if android)

No response

Android AAR (if android)

None

OpenCV version (if running on desktop)

No response

Describe the problem

I'm trying to customize the model type for `/mediapipe/tree/master/mediapipe/examples/ios/posetrackinggpu` example. If I understand correctly, posetrackinggpu supports 3 models: `pose_landmark_full`, `pose_landmark_heavy`, and `pose_landmark_lite`. I did attempt to change `//mediapipe/modules/pose_landmark:pose_landmark_full.tflite` to `//mediapipe/modules/pose_landmark:pose_landmark_heavy.tflite` or `//mediapipe/modules/pose_landmark:pose_landmark_light.tflite` in posetrackinggpu BUILD file as well as changing `model_complexity` to 0 or 2 in `mediapipe/python/solutsions/base.py` and recompiling with Basel. Changes seem to break pipeline, running app with new changes logs following error in the console: 

::Open() for node "poselandmarkgpu__poselandmarkbyroigpu__poselandmarkmodelloader__LocalFileContentsCalculator" failed: ; Can't find file: mediapipe/modules/pose_landmark/pose_landmark_full.tflite, GUSGoogleUtilStatusErrorKey=<GUSUtilStatusWrapper: 0x2835e80e0; status = Graph has errors:

Complete Logs

poselandmarkgpu__poselandmarkbyroigpu__poselandmarkmodelloader__LocalFileContentsCalculator" failed: ; Can't find file: mediapipe/modules/pose_landmark/pose_landmark_full.tflite, GUSGoogleUtilStatusErrorKey=<GUSUtilStatusWrapper: 0x2835e80e0; status = Graph has errors:

Metadata

Metadata

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions