- 博客(8)
- 收藏
- 关注
原创 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
179
原创 YOlOv8 ONNXRunTime-gpu 推理
(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的约瑟夫-雷德蒙(Joseph Redmon)和阿里-法哈迪(Ali Farhadi)开发。显示检测结果,获取检测目标框,输出检测框,打印检测标签,保存检测结果为.bmp(其它格式文件会降低算法执行保存速率,其它格式需要系统会进行压缩,影响执行速率)根据YOLOv8官方文档介绍,通过使用model.export()将.pt文件转为.onnx文件。将符合检测要求的目标框,添加至预测坐标中。将检测模型转到GPU中推理。
2024-05-14 11:50:31
947
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
2278
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人