Fig. 2

Illustration of the Non-Local, TPAVI, and TPAVI + structure. Note that we use the same color to represent the same type of feature map. The reshape operation is omitted for simplicity.
Illustration of the Non-Local, TPAVI, and TPAVI + structure. Note that we use the same color to represent the same type of feature map. The reshape operation is omitted for simplicity.