3_用YOLOv7训练自己的数据集——基于YOLOv7的跌倒检测_yolov7定位后怎么使用openpose跌倒识别

前两部分已经完成了环境部署与数据集，接下来开始进行模型训练。

1.选择训练模型

回到yolo_main工程目录下，在cfg/deploy下选择自己所要使用的模型，这里本人选择了yolov7x

打开yolov7x.yaml，修改参数，将nc设置成你要检测的类目。

 2.下载权重文件

在之前提到的github上mirrors / WongKinYiu / yolov7 · GitCode下载yolov7x的权重文件。

下载成功后将其放在weights文件夹下

 打开train.py，下拉找到代码里设置的命令行参数项，按照自己的情况进行修改。

上面这些设置好之后运行train.py，如果一切顺利的话，慢慢等训练完成就行了。如果你没有改代码其他设置，那么你可以在runs/train/exp文件夹下获取这次训练的有关信息，runs/train/exp/weights文件夹里面保存了这次训练结束后整个网络模型的权重，后面进行检测时会用到。