How to use the CBAM in yolo, could you give me an example?
lijoe123 opened this issue · 0 comments
lijoe123 commented
Prerequisite
- I have searched the existing and past issues but cannot get the expected help.
- I have read the FAQ documentation but cannot get the expected help.
- The bug has not been fixed in the latest version.
💬 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