4 个月前

TINYCD:一种(并非那么)深度的学习模型用于变化检测

TINYCD:一种(并非那么)深度的学习模型用于变化检测

摘要

本文介绍了一种轻量且高效的变更检测模型,称为TinyCD。该模型旨在满足工业需求,设计得比当前最先进的变更检测模型更快、更小。尽管其体积比对比的变更检测模型小13到140倍,并且计算复杂度至少降低了三分之一,但我们的模型在LEVIR-CD数据集上F1分数和IoU指标均优于当前最先进的模型至少1%,在WHU-CD数据集上的表现则超过8%。为了达到这些结果,TinyCD采用了Siamese U-Net架构,以全局时间维度和局部空间维度利用低级特征。此外,它采用了一种新的策略,在时空域中混合特征,不仅用于合并从Siamese主干网络获得的嵌入向量,还结合一个MLP模块,形成了一种新颖的空间-语义注意力机制——混合与注意力掩码块(Mix and Attention Mask Block, MAMB)。源代码、模型及实验结果可在此处获取:https://github.com/AndreaCodegoni/Tiny_model_4_CD

代码仓库

andreacodegoni/tiny_model_4_cd
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
building-change-detection-for-remote-sensingTinyCD
F1: 91.05
IoU: 83.57
Params(M): 0.28
building-change-detection-for-remote-sensing-1TinyCD
F1: 91.74
IoU: 84.74
change-detection-on-whu-cdTiny-CD
F1: 91.05
IoU: 83.57
Overall Accuracy: 99.10
Precision: 92.68
Recall: 89.47

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TINYCD:一种(并非那么)深度的学习模型用于变化检测 | 论文 | HyperAI超神经