DL00402-基于YOLOv5的旋转框目标检测
dota标注数据集格式转yolov5 rotation标注格式
脚本 utils/cut2rotation.sh ,修改数据集路径和滑窗切片大小及重叠面积即可
数据集路径修改
修改data文件夹下数据集定义
anchor聚类
cd utils,修改数据集路径,然后python autoanchors.py即可
模型anchor修改
将聚类的anchor更新到模型定义中 models/yolov5s_rotation.yaml
训练
脚本 dis_train_rotation.sh,修改数据路径运行
测试及评价
脚本 infer_remote_rotation.sh,修改数据路径及模型权重,会保存测试图像结果
ID:183500
详询客服 微信shujuqudong1 或shujuqudong6 或 qq68823886 或 27699885
图文详情请查看: http://matup.cn/731116609025.html