【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

你可能感兴趣的:(AI,人工智能,YOLO,pytorch)