search

kobiso / CBAM-keras

CBAM implementation on Keras
MIT License
359 stars 134 forks source link

add it to yolov4 #8

Open Lia1234567 opened 2 years ago

Lia1234567 commented 2 years ago

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

varungupta31 commented 2 years ago

@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.