logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

yolov8_obb旋转框不同类型位置信息的输出

然后,我的项目要求包括后续的机械臂抓取,坐标系是这么建立的:垂直线为0,水平线正方向为90度,水平线负方向为-90度,所以输出的角度为[-90,90],大家在得到四个顶点坐标后,可以按需求自己建立坐标系。angle = result.obb.xywhr[0][4].cpu().item()# 旋转角度(弧度)cx = result.obb.xywhr[0][0].cpu().item()# 中心点

文章图片
yolov8_obb旋转框不同类型位置信息的输出

然后,我的项目要求包括后续的机械臂抓取,坐标系是这么建立的:垂直线为0,水平线正方向为90度,水平线负方向为-90度,所以输出的角度为[-90,90],大家在得到四个顶点坐标后,可以按需求自己建立坐标系。angle = result.obb.xywhr[0][4].cpu().item()# 旋转角度(弧度)cx = result.obb.xywhr[0][0].cpu().item()# 中心点

文章图片
关于训练模型时num_workers和batch_size的设置匹配

我的想法是batch_size调低,gpu处理一次batch就更加轻松,对下一个batch的需求就更加渴望,那么负责dataloader的cpu就会不停歇地处理和加载数据喂给gpu。所以干脆把batch_size拉到很高,让gpu处理一批数据的速度不要太快,那么cpu加载数据也就更加游刃有余。我的cpu是i7,gpu是4090d,之前跑代码一直会蓝屏,按照网上的意见调低batch_size和num

文章图片
#batch#开发语言
到底了