Code and paper presentation do not match

[Ellen7ions]

The shape of query Q in Figure 3 is HWxC, but in the code

CamoFormer/model/decoder/decoder_p.py

Line 139 in 9f627e2

q = rearrange(q, 'b (head c) h w -> b head c (h w)', head=self.num_heads)

, the shape of query Q is CxHW.