自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(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关注的人

提示
确定要删除当前文章?
取消 删除