Advances, Systems and Applications
Table 1 The specific procedure of the ARM
Algorithm: Attention Refinement Module |
|---|
Input: input |
Output: x |
1. Apply Conv2d operation to input, resulting in input1 |
2. Apply AdaptiveAvgPool2d operation to input1, obtaining x |
3. Apply Conv2d operation to x. x = Conv2d(x, kernel_size = 1) |
4. Apply Sigmoid activation to x |
5. Element-wise multiply input1 by x |
6. Return x |