
SPARK-SAM 先适应红外目标域的 prompt-to-mask 响应,再用图像条件联合自提示实现自动分割,并量化两部分对精度和定位的贡献。

Z,自提示模块 Q 从 Z 预测 objectness、候选点偏移、候选分数、掩码质量和紧凑框。它从低分辨率 objectness map 取 Top-K 位置,用偏移修正坐标,再在坐标处双线性采样局部 token。P_I = Q(Z, R(Z_h)) = ({u_j}, b, T_res, {t_j})S_I = [P({u_j}, b), T_res, {t_j}]M_hat = D(Z, S_I)
A_p 产生软 mask、边界分布和质量估计。每个像素按三种关系分配权重:响应与标注对齐时为 ρ_a,背景上出现疑似假阳性时为 ρ_s,其余区域为 ρ_d,并满足 ρ_a > ρ_d > ρ_s。A_p 用 objectness、局部对比度、top-hat 响应、峰值锐度和高频证据生成候选先验;冻结响应再提供 mask contrast、面积兼容性、紧凑性和点—mask 中心一致性。最高候选提供训练点,多尺度框响应选择紧凑框或训练 mask 推导的 fallback。提示损失同时约束 objectness、point、box、ranking、coverage、candidate-mask quality 和 dense prior。1Z_h 预测残差:o_h = Up(o_l) + αR(Z_h)S_I 后,NUAA-SIRST、NUDT-SIRST 和 IRSTD-1K 的 mean nIoU 分别下降 28.27、8.84 和 18.68 个百分点。随机或打乱空间分支后,三套数据的 nIoU 都下降;框打乱的下降只有 0.11 到 0.20 个百分点。用测试参考 mask 推导的框替换预测框,IoU 分别增加 1.57、2.48 和 1.77 个百分点。1| 数据集 | SPARK-SAM IoU | 论文内排名 | 额外参数 |
|---|---|---|---|
| NUAA-SIRST | 75.78% | 14 个重训练 SAM 变体中第 2 | 0.726M 1 |
| NUDT-SIRST | 86.49% | 第 1 | 0.726M 1 |
| IRSTD-1K | 68.34% | 第 1 | 0.726M 1 |
| 数据集 | SPARK-SAM IoU | SCTransNet IoU | SPARK-SAM nIoU/F1/FA |
|---|---|---|---|
| NUAA-SIRST | 75.78% | 78.46% | 76.93/86.22/100.88 1 |
| NUDT-SIRST | 86.49% | 94.42% | 87.75/92.75/41.86 1 |
| IRSTD-1K | 68.34% | 69.89% | 64.47/81.19/46.02 1 |
1024² 输入的平均延迟为 31.451ms,中位数 30.716ms,P95 为 37.573ms,约 31.8 FPS;新增模块让平均延迟增加 0.051ms。这个基准保留了各方法的原生输入分辨率,FPS 不能脱离分辨率比较。1
arxiv.org
arxiv.org
github.com
This story was produced automatically by a channel. One sentence is all it takes for Neodrop to keep producing for you.