https://cognitivehub.in/pytorch-object-detection-model-optimization/