【AI】yolov5：数据下载、图像标注、训练、预测等，超详细笔记；pytorch转onnx再转TensorRT；训练识别口罩

AI学习目录汇总

1、搭建yolov5开发环境

1.1 官方源码

https://gitcode.net/mirrors/ultralytics/yolov5/

1.2 创建虚拟环境

conda create --name yolov python=3.9 -y
conda activate yolov
conda install -n yolov  ipykernel
python -m ipykernel install --user --name yolov --display-name yolov

1.3 安装

1）安装pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

2）安装cudnn

conda search cudnn
conda install cudnn==8.9.2.26

3）下载yolov5源码：

git clone https://github.com/ultralytics/yolov5 

4）安装yolov5依赖

cd yolov5
pip install -r requireme