Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Object detection model can't be used in C++ #5389

Open
ly0303521 opened this issue May 8, 2024 · 2 comments
Open

Object detection model can't be used in C++ #5389

ly0303521 opened this issue May 8, 2024 · 2 comments
Assignees
Labels
os:linux-non-arm Issues on linux distributions which run on x86-64 architecture. DOES NOT include ARM devices. platform:c++ Issues specific to C++ framework in mediapipe stat:awaiting response Waiting for user response type:modelmaker Issues related to creation of custom on-device ML solutions

Comments

@ly0303521
Copy link

Have I written custom code (as opposed to using a stock example script provided in MediaPipe)

None

OS Platform and Distribution

ubuntu20

Python Version

3.11

MediaPipe Model Maker version

0.2.1.3

Task name (e.g. Image classification, Gesture recognition etc.)

Object detection

Describe the actual behavior

model maker train is successful, but failed in C++ inference

Describe the expected behaviour

the trained model can inference in C++

Standalone code/steps you may have used to try to get what you need

In the begining, I use mediapipe follow the example "mediapipe/graphs/object_detection", its inferenced in C++, the model is "ssdlite_object_detection.tflite". 
Now, I tried retrain the object detection model, follow "https://developers.google.com/mediapipe/solutions/customization/object_detector", I found the models are 
MOBILENET_V2
MOBILENET_V2_I320
MOBILENET_MULTI_AVG
MOBILENET_MULTI_AVG_I384,
these are different from "ssdlite_object_detection.tflite".
I choosed "MOBILENET_V2" to try, training is successful, but I found MOBILENET_V2 model can't replace ssdlite_object_detection

Other info / Complete Logs

No response

@ly0303521 ly0303521 added the type:modelmaker Issues related to creation of custom on-device ML solutions label May 8, 2024
@ly0303521
Copy link
Author

#5308 more details in this issue I raised 1 month ago

@kuaashish kuaashish added os:linux-non-arm Issues on linux distributions which run on x86-64 architecture. DOES NOT include ARM devices. platform:c++ Issues specific to C++ framework in mediapipe labels May 8, 2024
@schmidt-sebastian
Copy link
Collaborator

Have you tried the model in MediaPipe Studio?

https://mediapipe-studio.webapps.google.com/studio/demo/object_detector

@kuaashish kuaashish added the stat:awaiting response Waiting for user response label May 16, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
os:linux-non-arm Issues on linux distributions which run on x86-64 architecture. DOES NOT include ARM devices. platform:c++ Issues specific to C++ framework in mediapipe stat:awaiting response Waiting for user response type:modelmaker Issues related to creation of custom on-device ML solutions
Projects
None yet
Development

No branches or pull requests

3 participants