Follow part 2 of this tutorial series to see how to train a specialized classification model for object localization using CNN's with PyTorch. Then see how to save and convert the model to ONNX.
Your post was dead. I vouched. You might want to email dang and ask why.