- 博客(3)
- 收藏
- 关注
RSS订阅原创 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关注的人