Hello, I’m attempting to get a custom object detection model running on a nano at 20-30 fps, with the highest accuracy possible. I’ve been following this guide:
https://www.dlology.com/blog/how-to-run-tensorflow-object-detection-model-on-jetson-nano/
and trying to work with ssd mobilenet v1/v2 and ssdlite mobilenet v2, from the model zoo here: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/tf1_detection_zoo.md
Using the method provided in the dlology link for converting to tensort and running the code, I get 8-10 FPS with ssd mobilenet v1, 6-8 FPS with ssdlite mobilenet v2, and ssd mobilenet v2 crashes the nano when I try to create the TRT graph (I think due to OOM). Is there a better “official” way of working with the tensorflow object detection API? Or is there a noob mistake of some sort I’m making here?
Thanks!
EDIT: all these models are running at 300x300 resolution