这个屌丝很懒，什么也没留下！

yolov5-obb 旋转目标检测从训练到部署_yolov5 obb训练

作者：AllinToyou | 2024-03-17 17:17:32

yolov5 obb训练

源码：yolov5-obb
训练的数据结构
数据格式
训练命令

python train.py   --weights 'weights/yolov5m.pt'   --data 'data/yolov5obb_demo.yaml'   --hyp 'data/hyps/obb/hyp.finetune_dota.yaml'   --epochs 10   --batch-size 64   --img 800   --device 3
1

-源码：yolov5-obb-tensorrt

数据格式应该为整型，不然训练会报错
模型训练的输入尺寸需被32整除，不然tensorrt推理会不对

本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】

yolov5-obb 旋转目标检测从训练到部署_yolov5 obb训练

1 模型训练

2. 模型tensorrt部署

`1 模型训练`

`2. 模型tensorrt部署`