yolov8训练自己的数据集以及权重转化为.engine格式(超详细)

1 获取.mp4

利用视频分析软件中的“数据处理”把采集的视频处理成.mp4文件，然后再放在另一个文件夹里面，如下：

2 pycharm切换文件

1 双击桌面pycharm

2 若是想要标记数据集：

(1)此时打开的pycharm项目是训练项目，就要重新打开最近的其他项目，操作如下：

a.鼠标点击File

b.鼠标点击Open Recent

c.鼠标点击 FileCaoZuo

点击之后会弹出，另一个项目的打开窗口，选择 New Window

这个时候任务栏下就会出现两个pycharm项目，如下：

选择的是 This Window 就会关闭当前窗口，任务栏只保留一个pycharm项目

（2）此时打开的正好是 FileCaoZuo 项目 ，继续向下操作。

2 .mp4文件提取帧图片

FileCaoZuo 项目里面的 pre_work 文件夹里面 “视频提取图片.py” .双击进去

1 把（1）处的视频路径换成刚刚另存文件夹的路径

2 （2）处的路径换成要保存图片的文件夹路径

3 右键运行

4 接着继续把1换成2、3、4、5、6，图片保存路径也换成2、3、4、5、6

3 帧图片筛选规则

1 把模糊不清、人不正常飞行的去掉

2 把相同动作的去掉

3 一套衣服训练集需要400-500张图片，共6个文件。每个文件夹筛选70-80张。

4 图片重命名

FileCaoZuo 项目里面的 pre_work 文件夹里面 “图片重命名.py” .双击进去

1 把路径换成刚刚保存的1文件图片的路径

2 i的值调整到0（0意思第一张图片从1开始命名）

3 右键运行

循环操作：

4 看1(2 3 4 5 )文件里面最后一张图片的名字 （注意顺序）

5 把路径换成刚刚保存的2(3 4 5 6)文件图片的路径

6 i的值依次查看上一个文件夹里面最后一张图片的名字

5 图片名指定长度

1 把6个文件的图片整合到一个新的文件夹(新文件命名随缘)里面的image里面

Eg：-新文件夹名

​ -image

​ -json

2 FileCaoZuo 项目里面的 pre_work 文件夹里面 “图片重名后指定长度.py” .双击进去

6 labelme使用规则

在 FileCaoZuo 项目 打开Terminal，输入labelme ，然后Enter

会出现新的界面，如下：

标注界面介绍以及把人框出来

标注关节点以及删除标错或者多标关节点

关节点标注之后保存到json文件里面

7 标注人体关节点

关键点标注原则：

（1） 按照下图的顺序进行标注

（2） 若是某个点被遮挡或者看不清就跳过该关节点的标注

8 检测关键点标注的准确性

FileCaoZuo 项目里面的 pre_work 文件夹里面 “coco_json.py” .双击进去

把如下json文件的路径写到“coco_json.py”里面的1中；2换成也换成同样的路径，保留路径后面的\val.json

FileCaoZuo 项目里面的 pre_work 文件夹里面 “检查coco格式的json.py” .双击进去

把路径换成image、json里面的val.json

查看图片有没有明显标注错误。

9 json文件转化成txt文件

在第5步里面“新建的文件夹”里面再创建一个 “annotations”

FileCaoZuo 项目里面的 pre_work 文件夹里面 “pose_json_txt.py” .双击进去

修改1 2处的路径，运行这个文件

10 划分数据集、验证集

在第5步里面“新建的文件夹”里面再创建一个 “split”

FileCaoZuo 项目里面的 pre_work 文件夹里面 “seg_split_data.py” .双击进去

修改1 2 3处的路径，运行这个文件

到此数据集就准备好了，准备好的文件如下：

1 数据集路径：D:\Data

2 点开进去之后： D:\Data\mi_pose

split文件夹里面就是划分好的数据集

2 pycharm切换文件

1 双击桌面pycharm

2 若是想要标记数据集：

（1）此时打开的正好是 Ultralytics-new 项目 ，继续向下操作。

（2）此时打开的pycharm项目是数据集项目，就要重新打开最近的其他项目，操作如下：

a.鼠标点击File

b.鼠标点击Open Recent

c.鼠标点击 Ultralytics-new

点击之后会弹出，另一个项目的打开窗口，选择 New Window

这个时候任务栏下就会出现两个pycharm项目，如下：

选择的是 This Window 就会关闭当前窗口，任务栏只保留一个pycharm项目

11 调整训练参数

在Ultralytics-new 项目中打开ultralytics中的cfg中的datasets中的coco8-pose.yaml文件

把split里面对应的路径换成下面

在Ultralytics-new 项目中打开ultralytics中的models中的yolo中的pose中的train.py文件,运行

运行之后会在pose目录下生成一个run文件，里面的train文件里保存的就是训练的结果。最下面的训练结果就是最新的。

12 .pt文件转化.onnx

在 Ultralytics-new 项目中 打开Terminal，输入下面命令：

yolo export model=r’D:\yolov8\ultralytics-new\ultralytics\models\yolo\pose\runs\pose\train6\weights\best.pt\best.pt’ format=onnx

这里的best.pt文件路径换成自己的。

结果会在当前.pt同级目录下生成.onnx

13 .onnx转化.engine

将生成的best.onnx文件（这里best可以重命名）复制到C盘里面的TensorRT-8.4.1.5中的bin目录下。

路径：C:\tensorrt\TensorRT-8.5.2.2\bin

然后 输入cmd进入命令行模式：

trtexec.exe --onnx=./best.onnx --saveEngine=./best.engine (若是上面的best换名字了这里要对应上)

………等一段时间以后，会在bin文件里面生成一个对应的.engine文件

14 调换权重路径

这里把视频分析软件里代码的权重路径改一下。