mit-han-lab / efficientvit

EfficientViT is a new family of vision models for efficient high-resolution vision.
Apache License 2.0
1.79k stars 164 forks source link

MSA只有5x5卷积算子 #12

Closed violet2422 closed 1 year ago

violet2422 commented 1 year ago

https://github.com/mit-han-lab/efficientvit/blob/master/efficientvit/models/nn/ops.py 为什么和论文图2不一样

han-cai commented 1 year ago

Thanks for your interest in our work.

Figure 2 is an illustration of the general idea. As discussed in the implementation details, we adopt a two-branch design in MSA for a better trade-off between accuracy and efficiency.