Code can be found on Github. Or a side-by-side comparison of the networks here.
I re-trained the SSD-Mobilenet-v1 checkpoint provided by jetson-inference with a set of 2500 images that I labeled with labelImg.
These kittens were living with me for a few weeks, so I had no choice but to train a model to tell them apart! I enjoyed the experience of training my first model using my own data to provide inference for a task in my real world.
I appreciate any tips or comments on my work, as I am just getting started in this awesome field!