-
Notifications
You must be signed in to change notification settings - Fork 70
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
Custom model integration #40
Comments
I found three scripts that include the word 'Tracking': camera.py, ssd_mobilenet_v3_coco.py, visualization.py |
Hey @jp3spinelli! You'll need to create a class similar to For I'm also using the Here's how I add this op with an older TFlite toco converter. Word of warning, this might not work with the newer MLIR-based TFlite converter. After you have a working model class setup, you can call You'll probably have to adjust the default PID gains, since these are tuned to track a human-sized object from a camera at roughly desk height. My notes on PID controllers are below as well. Good luck! |
Closing for now, so it's easier for me to see issues in need of triage at a glance. I'd love to see your final setup when you get this working! |
I am trying to use this to get my pan/tilt hat to track an opposing Raspberry Pi camera. I have trained my custom model and converted it to TFlite, I just need to know how to integrate my model into this repository to be able to track. I trained my model on ssd_mobilenet_v3_small_coco_2019_08_14.tar.gz, which you use, so I figure there is some way to tweak the files to make this work.
Thanks!
The text was updated successfully, but these errors were encountered: