title: Retraining Single Shot MultiBox Detector model on a custom data set?
body: <p>Has anyone had any success retraining one of pytorch's saved SSD models using a custom dataset? I'm having a hard time finding documentation about what the input images need to look like, target dictionary needs to contain, etc.</p>
  <p>Am I overlooking documentation somewhere?</p>
  <p>This is the model category I am talking about: <a href="https://pytorch.org/hub/nvidia_deeplearningexamples_ssd/" rel="nofollow noreferrer">https://pytorch.org/hub/nvidia_deeplearningexamples_ssd/</a></p>
  <p>The pytorch model I've been looking at retraining is <code>torchvision.models.detection.ssdlite320_mobilenet_v3_large</code>. However, I am open to other models.</p>
  <p>I have a relatively simple problem in which I want to identify people's hands in images. I'm not sure what all needs to be passed in, and I'm having a hard time figuring it out.</p>
  <p>Any advice or high quality example code would be appreciated. My goal is to really learn what is going on under to hood rather than just copying code.</p>