Note, requires pytorch and ultralytics packages:

• https://github.com/ultralytics/ultralytics
• https://pytorch.org/get-started/locally/