这里写自定义目录标题
在YOLOv5中添加CBAM报错:RuntimeError: Given groups=1, weight of size [3, 128, 2, 2], expected input[1, 48, 128, 128] to have
修改YOLOv5要修改3个脚本,(1)common.py文件中定义模块。(2)yolo.py中添加到parse_model函数下的一个列表中以解析模块。(3).yaml文件中使用模块。
我的问题出在第二步自定义的模块没有添加到列表中,故输出的tensor报错。