Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add it to yolov4 #8

Open
Lia1234567 opened this issue Mar 2, 2022 · 1 comment
Open

add it to yolov4 #8

Lia1234567 opened this issue Mar 2, 2022 · 1 comment

Comments

@Lia1234567
Copy link

may i know how can i add CBAM attention to yolov4?

@varungupta31
Copy link

@Lia1234567 I think you need to decide where it is to be put, i.e. between which layers. As per the paper, bottleneck regions would be a good starting point. Depending on your YOLOv4 repo, start by identifying where the architecture is defined, simply import the module provided in this repo, and place the CBAM block.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants