Tensorflow ROS node for Nvidia jetson tx2

asked 2018-07-31 18:29:37 -0500

AutoCar gravatar image

Is there a ROS node wrapper for Tensorflow for inference? Ideally the node is targeting nvidia jetson tx2 jetpack 3.1. I have build and trained a model on a desktop. Now I would like to run inference on my robot which is driven by jetson tx2 and ZED camera.

edit retag flag offensive close merge delete

Comments

You probably will have to write your own, but it's usually relatively simple. I've done it for Caffe, first as a Python node (actually a pain as Python doesn't thread too well) then as C++. It just takes sensor_msgs::Image messages as input, then outputs the results of the DL as a custom message.

KenYN gravatar image KenYN  ( 2018-07-31 21:02:42 -0500 )edit
1

Does Announcing the tensorflow_ros package: the easy way to get to C++ Tensorflow API in ROS sound like something you could use? It doesn't target Jetsons specifically though.

gvdhoorn gravatar image gvdhoorn  ( 2018-08-01 02:22:12 -0500 )edit