I have been trying to use DIGITS to use on our Jetson Nano for quite some time now to help us get Detect-net models to run for image detection. I have run into a lot of errors with everything even by following the Jetson-Inference tutorial to a T. Is there any other programs that are better? We are using the current version of the Jetpack and we need a program without WIFI so we have to be able to run the tracking program locally. Any help would be appreciated!
We want to be able to track an object with a camera while the camera is moving and know exactly where in the image it is. And do you know if there is any documentation on how to collect your own datasets, cause we only want it to train onto our own pictures of balls. Thanks!!
Hi dusty-NV, we already collected our own datasets and got the image-net working to see out ball but we wanted to use detect net so we could get the position of the ball on the screen if there is an easier way to do this please let me know :)
If your ball is always visually distinctive enough from your background (a specific hue, for example), you might be able to use a simple blob detection/tracking library to do what you want (think PS Eye). It all depends on your use case.
The SSD-Mobilenet and SSD-Inception models are currently trained with TensorFlow (which is why they do not appear yet in the training portion of the Hello AI World tutorial, which uses PyTorch). So you could re-train these models on your dataset with the TensorFlow Object Detection API (ideally running on a PC/server/cloud instance, or it might run on the Nano - albeit slowly and with extra swap space). Here are some resources about that:
You will have to build and install OpenCV first for that specifically. You can search the forum for threads providing scripts that will do that for you.
Dusty_nv may have some better suggestions for such a simple use case. If you google CUDA blob detection you are likely to find many other code samples and suggestions. There are many approaches to this problem.
First of all: I very much appreciate the work and time you are putting into the community; the last months I’ve learned a lot reading your and other people’s directions in relation to the Jetson nano.
Second, unfortunately I have not yet succeeded in training my own object detector using the nano and Tensorflow. Besides the above links, I’ve followed many more tutorials and (re-)flashed my sd card (64GB to avoid memory issues) over and over again to sure that I began from scratch. I’ve got the latest sd card images for the nano but I seem to keep stumbling when I want to start the training (tensor flow model_train.py fails with message: “import tensorflow.contrib.slim as slim ModuleNotFoundError: No module named ‘tensorflow.contrib’”); When I search the net on this message it appears that the contrib module was removed in tensorflow 2; with the 1.xx tensorflow versions I did not have success as well unfortunately.
Maybe I am suffering from “tunnel vision” by now but I have not been able to find 1 clear step-by-step tutorial to get the training with my custom images going. To summarise: I’ve got my image dataset (dice, showing either the 1 or 6), I’ve labelled the images (using LabelImage), I’ve created the TFRecord files for training and testing, I’ve installed tensorflow, protobuf, cocoapi, etc successfully, sufficient swap memory available…and then the above error shows.
Can you, or anybody else reading this, please point me to a A-to-Z tutorial that leads me to creating a trained object detector for my custom images?
Hi, I really want to train the SSD mobile net on the nano. Because
The dusty_nv tutorial is easy to follow. :>
I only have a window PC and did try a virtual machine Linux but there are errors and library dependency issue using DIGITS.
I have my model trained in Azure and it has several output formats, onnx, docker, tensorflow, how can I import my model into nano?
If I train in, for example, tensorflow on window, I can copy the model to nano without any modification, right? I couldn’t find an official tensorflow example of training SSD mobilenet V2,https://www.tensorflow.org/tutorials/, is there any link I can follow?