yolov5 pytorch复现

要复现Yolov5 PyTorch项目，你可以按照以下步骤进行操作：
1. 首先，在主项目目录下运行以下命令来安装所需的依赖库：

   pip install torch==1.8.0 cu111 torchvision==0.9.0 cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

2. 运行以下命令来进行模型训练和推理预测：

   python yolov5/train.py --weights '' --cfg yolov5s.yaml --data coco.yaml --epochs 300
   python yolov5/detect.py --source data/images --weights runs/train/exp/weights/best.pt

3. 如果你想要导出模型为ONNX格式，可以运行以下命令：

   python yolov5/export.py --weights runs/train/exp/weights/best.pt --img 640 --batch 1

4. 最后，你可以使用导出的模型进行验证：

   python yolov5/val.py --data coco.yaml --weights yolov5s.onnx --iou 0.65