Serge-weihao / CCNet-Pure-Pytorch

Criss-Cross Attention (2d&3d) for Semantic Segmentation in pure Pytorch with a faster and more precise implementation.
MIT License
183 stars 21 forks source link

为什么CCNet输入和输出总是一样的呢? #21

Open lijing-coder opened 1 year ago

lijing-coder commented 1 year ago

图片

Serge-weihao commented 9 months ago

https://github.com/Serge-weihao/CCNet-Pure-Pytorch/blob/ff51a27bc5e111b2e7ddc9258266a2fea4637ac6/networks/CC3d.py#L98 https://github.com/Serge-weihao/CCNet-Pure-Pytorch/blob/ff51a27bc5e111b2e7ddc9258266a2fea4637ac6/networks/CC3d.py#L25

lijing-coder commented 9 months ago

大佬,还有一个问题,麻烦解答一下,论文中给出了模型的flops和memory,但是没有给params指标,params和non-local相比会不会也有所降低?具体降低多少呢?