XiaLiPKU / EMANet

The code for Expectation-Maximization Attention Networks for Semantic Segmentation (ICCV'2019 Oral)
https://xialipku.github.io/publication/expectation-maximization-attention-networks-for-semantic-segmentation/
GNU General Public License v3.0
680 stars 130 forks source link

FLOPs #46

Closed songqi-github closed 4 years ago

songqi-github commented 4 years ago

How to compute the FLOPs and parameters of your EMA module? Could you please share the computing details? Thanks!

XiaLiPKU commented 4 years ago

We adopt the same rule in torchstat. That is, for matrix computation mn x np, the FLOPS here is mnp