【AI】yolov5：数据下载、图像标注、训练、预测等，超详细笔记；pytorch转onnx再转TensorRT；训练识别口罩 - 大白社区
- 发帖时间：
- 2023-07-08 19:36:58
摘要：Yolov5入门操作超详细笔记分享，包括数据下载、图像标注、GPU训练、预测等权重转换：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