复现YOLO-of-RoboMaster-Keypoints-Detection-2023

发布时间 2023-11-29 00:29:22作者: Tanگگ

开源仓库地址:

https://github.com/zRzRzRzRzRzRzR/YOLO-of-RoboMaster-Keypoints-Detection-2023

该仓库提供了数据集,目前只是测试了能不能训练

复现过程

git clone

下载源码

下载相应权重文件yolov7.pt

去到yolov7开源仓库,找到release,下载yolov7.pt

下载数据集并设置yaml

下载完后,将
armor_kpt.yaml修改,将train和val给改成自己的目录

尝试首先测试测试装甲版四点模型


将train.py文件修改成这样,(把视觉版识别模型注释,把装甲板四点模型注释取消并修改)

训练

终端输入:

python train.py