May have already done this? If so, make so have notes. https://medium.com/cvrealtime/from-pytorch-yolo-to-onnx-a-computer-vision-engineers-guide-to-model-optimization-bfdf86ae36f4