Ultralytics THOP package for fast computation of PyTorch model FLOPs and parameters.
pip install ultralytics-thop