- 博客(14)
- 收藏
- 关注
RSS订阅原创 YOLOv8添加CBAM注意力机制（小白修改）
def parse_model(d, ch, verbose=True): # 中添加CBAM模块。在YOLOv8中ultralytics->nn->tasks.py文件，导入CBAM模块，在。修改配置文件，复制一份yolov8.yaml文件，命名位yolov8_CBAM.yaml文件。from ultralytics.nn.modules import # 添加CBAM模块。查看__init__.py文件中CBAM模块。查看conv.py模块是否存在“CBAM”新建文件查看模块是否添加成功。
2024-07-19 09:43:10
997
原创 YOlOv8 ONNXRunTime-gpu 推理
(You Only Look Once）是一种流行的物体检测和图像分割模型，由华盛顿大学的约瑟夫-雷德蒙（Joseph Redmon）和阿里-法哈迪（Ali Farhadi）开发。显示检测结果，获取检测目标框，输出检测框，打印检测标签，保存检测结果为.bmp（其它格式文件会降低算法执行保存速率，其它格式需要系统会进行压缩，影响执行速率）根据YOLOv8官方文档介绍，通过使用model.export()将.pt文件转为.onnx文件。将符合检测要求的目标框，添加至预测坐标中。将检测模型转到GPU中推理。
2024-05-14 11:50:31
1784
1
原创 yolov5 onnx推理报错
onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: images for the following indices index: 0 Got: 1 Expected: 16 Please fix either the inputs or the model.
2023-11-08 11:33:49
3277
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人