实践教程｜YOLOv8 OBB实现自定义旋转对象检测

文章预览

↑ 点击 蓝字  关注极市平台 作者丨 gloomyfish 来源丨OpenCV学堂 编辑丨极市平台 极市导读   本文从 准备数据集、训练模型，导出与测试和部署推理四个方面， 介绍了如何使用YOLOv8 OBB实现自定义旋转对象的检测。   >> 加入极市CV技术交流群，走在计算机视觉的最前沿 数据集制作 我用手机拍了一张图像 然后自己写个代码，每旋转一度保存一张图像，这样就成功生成了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 p ………………………………