YOLOv5系列（2）——YOLOv5导出jit，onnx，engine

做个YOLOv5的专题，这部分写一些YOLOv5的工程部署方面的问题，持续更新。

一、YOLOv5导出jit

YOLOv5自导出，我们可以直接用它的导出代码：models/export.py

"""Exports a YOLOv5 *.pt model to ONNX and TorchScript formats

Usage:
    $ export PYTHONPATH="$PWD" && python models/export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
"""

import argparse

import torch

from utils.google_utils import attempt_download
from utils.general import set_logging

if __name__