yolov5

搭建python环境

yolov5环境搭建教程

yolov5框架下载

使用教程

看到项目第一眼会很懵，我把他整理了一下放到了xunlianji这个文件夹中

data--取自主目录data下的，是用来存储训练集、类型、还有验证集路径的地方

images--忘记了应该是没用的

models--取自主目录models下的，models是模型文件夹，里面主要是一些网络构建的配置文件和函数，其中包含了该项目的四个不同版本，分别是 s,m,l,x从名字就可以看出，模型的大小，它们的检测速度分别是从快到慢，但是精度是从低到高，根据训练自己的数据集，就需要修改这里面相对应的yaml文件来训练自己模型

pt--这个是用来导出onnx的地方，把生成的pt文件放在这里面，执行命令会导出到这个目录下

shujuyuan--这个是导入训练集的地方，可以使用T-Rex Lable进行标注，不用安装挺好的

test.py--检测环境是否正常，有无独立显卡

train.py--开始训练模型，并导出pt文件

detect.py--验证模型

export.py--把pt文件转为onnx文件

当所有都准备就绪后，执行test查看环境

 执行train.py开始训练，并导出pt文件，会在主目录-runs-train-exp某某-weights-best.pt，需要把他拷贝到主目录-xunlianji-pt目录下

 执行export.py，把shujuyuan-pt目录下的best.pt，导出onnx文件，存储位置跟pt在一起。下面有红色的，不是报错。