
PICANet 利用 24 方向多尺度非可训练深度卷积解耦红外弱小目标的高斯向心物理先验,并通过双向门控融合与级联非对称交叉注意力阻断跳连杂波,在三套公开基准上显著超越现有 SOTA 模型。



| 方法 | 类型 | NUDT-SIRST mIoU / Fa | IRSTD-1k mIoU / Fa | SIRST-Aug mIoU / Fa | 参数量 (M) | FLOPs (G) | GPU 耗时 (s) |
|---|---|---|---|---|---|---|---|
| Top-hat | 模型驱动 | 22.36 / 771.20 | 5.59 / 102.30 | 16.39 / 166.30 | – | – | 0.0018 (CPU) 1 |
| IPI | 模型驱动 | 34.83 / 106.90 | 18.67 / 93.37 | 21.90 / 93.17 | – | – | 3.0972 (CPU) 1 |
| MPCM | 模型驱动 | 25.96 / 316.50 | 14.81 / 64.24 | 19.49 / 304.80 | – | – | 0.0624 (CPU) 1 |
| ISNet | 数据驱动 | 87.77 / 9.42 | 64.73 / 15.03 | 70.12 / 21.57 | 0.97 | 30.62 | 0.0173 1 |
| UIUNet | 数据驱动 | 88.71 / 8.91 | 63.06 / 20.57 | 71.80 / 28.29 | 50.54 | 54.42 | 0.0317 1 |
| SCTransNet | 数据驱动 | 91.06 / 11.19 | 64.38 / 27.46 | 72.98 / 38.75 | 5.74 | 20.24 | 0.0213 1 |
| L²SKNet | 数据驱动 | 93.26 / 5.19 | 67.81 / 27.46 | 73.97 / 55.10 | 1.07 | 6.01 | 0.0122 1 |
| RPCANet | 算法展开 | 89.31 / 28.72 | 63.21 / 43.90 | 72.54 / 34.14 | 0.68 | 27.48 | 0.0096 1 |
| DRPCANet | 算法展开 | 93.12 / 19.50 | 63.93 / 49.20 | 73.93 / 30.45 | 1.17 | 91.05 | 0.0101 1 |
| CTVNet | 算法展开 | 92.05 / 8.97 | 66.35 / 20.47 | 72.96 / 38.79 | 14.90 | 85.74 | 0.0195 1 |
| L-RPCANet | 算法展开 | 94.39 / 1.39 | 67.44 / 43.90 | 75.35 / 26.43 | 0.22 | 8.59 | 0.0029 1 |
| PICANet (U) | 物理启发 | 95.19 / 1.28 | 70.31 / 23.38 | 75.48 / 28.56 | 6.33 | 8.53 | 0.0129 1 |
| PICANet (F) | 物理启发 | 94.69 / 1.34 | 69.35 / 29.43 | 75.58 / 25.95 | 2.74 | 4.27 | 0.0089 1 |

| 骨干配置 | 物理先验组合 | NUDT mIoU / Fa | IRSTD-1k mIoU / Fa | SIRST-Aug mIoU / Fa |
|---|---|---|---|---|
| U-Net 基线 | 无先验 | 86.32 / 18.65 | 59.12 / 60.38 | 63.41 / 83.36 1 |
| U-Net | + PCP1 | 89.29 / 8.41 | 63.66 / 48.56 | 66.83 / 48.89 1 |
| U-Net | + PCP2 | 89.52 / 9.52 | 64.11 / 50.19 | 67.22 / 50.08 1 |
| U-Net | + PCP1 & PCP2 (简单拼接) | 90.78 / 10.56 | 64.78 / 40.81 | 69.54 / 40.65 1 |
| U-Net | + PCPs & BPGFE (完整融合) | 95.19 / 1.28 | 70.31 / 23.38 | 75.48 / 28.56 1 |
| ResNet-FPN 基线 | 无先验 | 85.16 / 28.33 | 58.52 / 62.45 | 61.78 / 81.97 1 |
| ResNet-FPN | + PCPs & BPGFE (完整融合) | 94.69 / 1.34 | 69.35 / 29.43 | 75.58 / 25.95 1 |
conda create -n picanet python=3.10 -y
conda activate picanet
pip install -r requirements.txtdata/ 结构:data/
├── NUDT-SIRST/
├── IRSTD-1k/
└── SIRST-Aug/python train_UNet.pypython train_FPN.py
训练损失函数结合了多尺度二值交叉熵 、平滑交并比 与目标邻域掩码均方误差 。其中 BCE 权重 随训练进程由大变小,聚焦初始定位;IoU 权重 随进度增大,在后期强化精准分割;先验约束权重 全程固定为 1.0。1python t_models.py 得到 mIoU、F1、Pd 与 Fa 指标。所有训练与评估在一台配置 Intel Xeon Gold 6526Y CPU 与 NVIDIA RTX 4090 GPU 的工作站上完成。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.