开源仓库地址:
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