pytorch yolov5 训练自定义数据

pytorch yolov5 训练自定义数据

---

---

前言

环境

python: 3.9.7

torch: 1.10.2

labelimg: 1.8.6

#yolov5 https://github.com/ultralytics/yolov5
#pytorch https://pytorch.org/
#labelimg https://github.com/tzutalin/labelImg

paddleocr 有三种模型 det 检测 cls 方向 rec 识别

---

一、创建环境

1. 安装miniconda

   https://blog.csdn.net/mtl1994/article/details/114968140
   

2. 创建环境

   #linux 需要先 source
   conda create -n pytorch_yolov5 python=3.9.7  --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
   

3. 安装完以后进入环境

   conda activate pytorch_yolov5
   

二、安装环境

1. pytorch

   #选择对应的cuda/cpu版本
   pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/wh