I have a Jetson Nano and I plan to train an Object Detection Model on it.
I have seen the new documentation on how to use PyTorch and Resnet-18 over here: https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-transfer-learning.md
I have also seen the following resources. They are very similar as they use the ant and bee datasets:
I have tried the zaferarican site first. The training works as advertised but it doesn’t tell me how to save or where the new model is saved.
All of the samples above are for training Image Classification models.
However, I’m looking to do Transfer Learning on an Object Detection Model.
Based on my previous attempts at Training, the main difference in Training Object Detection Models is that I also add a folder where the coordinates are in each of the images in my train and evaluate/test folders. However, aside from that, I don’t know how to proceed as I’ve only used a script to do the actual heavy lifting.
Is there a guide or tutorial on how to do Transfer Learning with Object Detection on a Jetson Nano using PyTorch? Any leads will be very much appreciated.