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.