- 博客(4)
- 问答 (1)
- 收藏
- 关注
RSS订阅原创 yolov8_obb旋转框不同类型位置信息的输出
然后，我的项目要求包括后续的机械臂抓取，坐标系是这么建立的：垂直线为0，水平线正方向为90度，水平线负方向为-90度，所以输出的角度为[-90,90]，大家在得到四个顶点坐标后，可以按需求自己建立坐标系。angle = result.obb.xywhr[0][4].cpu().item() # 旋转角度（弧度）cx = result.obb.xywhr[0][0].cpu().item() # 中心点 x。h = result.obb.xywhr[0][3].cpu().item() # 高度。
2024-09-27 14:57:12
1651
4
原创 关于训练模型时num_workers和batch_size的设置匹配
我的想法是batch_size调低，gpu处理一次batch就更加轻松，对下一个batch的需求就更加渴望，那么负责dataloader的cpu就会不停歇地处理和加载数据喂给gpu。所以干脆把batch_size拉到很高，让gpu处理一批数据的速度不要太快，那么cpu加载数据也就更加游刃有余。我的cpu是i7，gpu是4090d，之前跑代码一直会蓝屏，按照网上的意见调低batch_size和num_workers。发现batch_size越低，cpu占用率越高，而gpu占用率百分之一都不到。
2024-07-09 09:11:00
388
原创 求助：RuntimeError: CUDA error: unknown error
以前能跑完训练的代码，上周开始跑到一半就会报错，batch_size调小了一半也不行。
2024-07-08 10:23:09
636
1
空空如也
Yolov5-in-Deepstream-5.0编译失败
2021-09-17
TA创建的收藏夹 TA关注的收藏夹
TA关注的人