Open Lia1234567 opened 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.
may i know how can i add CBAM attention to yolov4?