Install yolort

Required

Above all, follow the official instructions to install PyTorch 1.8.0+ and torchvision 0.9.0+.

To install yolort, you may either use the repository

and setup.py from there or use

pip install -U yolort

to get the latest release and

pip install 'git+https://github.com/zhiqwang/yolov5-rt-stack.git'

to install the development version.

Optional

Install pycocotools (for evaluation on COCO):

pip install -U 'git+https://github.com/ppwwyyxx/cocoapi.git#subdirectory=PythonAPI'