- 博客(3)
- 收藏
- 关注
原创 tensorrt推理时GPU专用显存一直增加
针对上述问题,检查将数据存入显存后是否及时释放了。经过检查添加这两个语句即可解决。求没bug,顺顺利利!
2024-03-01 15:34:39
444
1
原创 yolov8 tensorrt相关问题
1、Yolov8在python端导出tensorrt模型,不能直接用于C++,是因为读取文件的格式有差异,(应该是可以解决的,只是我的导出问题)。目前找到了在python端的解决方法,把yolov8-main/ultralytics/nn/autobackend.py文件中的186行左右,注释这两行代码,就可以运行在windows中导出的engine文件了。最好的方法是,在windows中用什么显卡跑engine模型,就用什么显卡导出,否则会出现显卡架构差异,导致两边都不可以读取。
2024-03-01 09:47:36
481
1
原创 yolov5模型数据集划分后存放目录展示
读了dataloader代码才知道,两种划分方式都可以,读标签的实质是将images替换为labels。
2023-03-23 14:10:13
153
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人