open-mmlab/mmyolo

How to use the CBAM in yolo, could you give me an example?

lijoe123 opened this issue · 0 comments

Prerequisite

💬 Describe the reimplementation questions

Could you give me an example of how to use the CBAM in yolo, i have no idea of how to write the config just like the GeneralizedAttention (你好,请问您可以教一下怎样在yolo使用CBAM吗?我在您的文章的CBAM链接里没有看到相关参数设置的介绍。):
base = './yolov5_s-v61_syncbn_8xb16-300e_coco.py'

model = dict(
    backbone=dict(
        plugins=[
            dict(
                cfg=dict(
                    type='GeneralizedAttention',
                    spatial_range=-1,
                    num_heads=8,
                    attention_type='0011',
                    kv_stride=2),
                stages=(False, False, True, True))
        ]))

Environment

The environment is ok

Expected results

I hope i can use the CBAM in the yolo

Additional information

No response