轻量级模型MobileSAM:处理一张图像仅需10ms,比FastSAM快4倍

轻量级模型MobileSAM:处理一张图像仅需10ms,比FastSAM快4倍

arXiv上的新论文MobileSAM提出了一个轻量级而又高性能的分割模型,适用于移动端部署。

800_auto

原始的SAM模型采用的ViT-H解码器计算量大,而MobileSAM基于“解耦蒸馏”方法,使用只有原生SAM 60倍小的参数TinyViT解码器,在保持SAM效果的同时极大缩减模型规模。

MobileSAM通过优化训练方法,只需要使用1%的数...

点击查看剩余70%

{{collectdata}}

网友评论