【深度学习】实践教程｜YOLOv8 OBB实现自定义旋转对象检测

文章预览

作者丨 gloomyfish    来源丨OpenCV学堂     编辑丨极市平台 导读   本文从 准备数据集、训练模型，导出与测试和部署推理四个方面， 介绍了如何使用YOLOv8 OBB实现自定义旋转对象的检测。   数据集制作 我用手机拍了一张图像 然后自己写个代码，每旋转一度保存一张图像，这样就成功生成了360张图像及其注释文件，分为训练集与验证集。训练文件夹包含 320张带有注释的图像。测试和验证文件夹都包含 40 张带有注释的图像。数据集部分图像显示如下： 模型训练 准备好数据集以后，直接按下面的命令行运行即可： yolo obb train data=pen_dataset.yaml model=yolov8s-obb.pt epochs=25 imgsz=640 导出与测试 模型导出与测试 # export model yolo  export  model=yolov8s-obb.pt format=onnx # inference model yolo obb predict model=pen_best.pt  source =pen_rotate_test.png 部署推理 转成ONNX格式文件以后， ………………………………