RealScene-ISTD
收藏资源简介:
RealScene-ISTD是一个用于红外小目标检测的真实场景基准数据集,包含图像和掩码文件,以及训练和测试的索引文件。
RealScene-ISTD is a real-world benchmark dataset for infrared small target detection, comprising image and mask files as well as training and testing index files.
RealScene-ISTD 数据集概述
数据集简介
- 标题: Rethinking Generalizable Infrared Small Target Detection: A Real-scene Benchmark and Cross-view Representation Learning
- 概述: 该数据集专注于红外小目标检测的通用性研究,提供了一个真实场景的基准数据集和跨视图表示学习方法。
数据集结构
-
下载地址: Baidu Drive
-
目录结构:
├──./datasets/ │ ├── RealScene-ISTD │ │ ├── images │ │ │ ├── 000001.png │ │ │ ├── 000002.png │ │ │ ├── ... │ │ ├── masks │ │ │ ├── 000001.png │ │ │ ├── 000002.png │ │ │ ├── ... │ │ ├── img_idx │ │ │ ├── train_RealScene-ISTD.txt │ │ │ ├── test_RealScene-ISTD.txt
代码演示
-
Gamma_Correction: bash python Dataset_Alignment/gamma_correction.py
-
Topk_Fusion: bash python Dataset_Alignment/Topk_Fusion.py
-
小目标补丁下载: Baidu Drive
使用方法
-
训练: bash python train.py
-
测试: bash python test.py
定量结果
RealScene-ISTD测试结果
| Method | mIoU (x10(-2)) | Pd (x10(-2)) | Fa (x10(-6)) |
|---|---|---|---|
| ACM-Net | 64.25 | 91.97 | 250.79 |
| ALC-Net | 68.88 | 90.91 | 146.85 |
| DNA-Net | 73.79 | 91.33 | 60.06 |
| RDIAN | 59.46 | 90.70 | 357.92 |
| ISTDU-Net | 74.40 | 93.87 | 75.35 |
| UIU-Net | 31.75 | 89.43 | 2928.64 |
| SCTransNet | 75.01 | 94.29 | 53.59 |
| Ours | 79.32 | 96.83 | 5.40 |
IRSTD-1K测试结果
| Method | mIoU (x10(-2)) | Pd (x10(-2)) | Fa (x10(-6)) |
|---|---|---|---|
| ACM-Net | 57.68 | 92.23 | 115.28 |
| ALC-Net | 62.69 | 92.28 | 58.99 |
| DNA-Net | 64.14 | 95.64 | 40.90 |
| RDIAN | 60.12 | 93.96 | 98.88 |
| ISTDU-Net | 63.20 | 96.98 | 14.36 |
| UIU-Net | 39.10 | 92.62 | 7008.86 |
| SCTransNet | 66.93 | 93.96 | 15.45 |
| Ours | 72.44 | 95.30 | 14.08 |
权重下载
- RealScene-ISTD权重: Baidu Drive
- IRSTD-1K权重: Baidu Drive
致谢
- 基于 SCTransNet 构建
- 对比实验基于 BasicIRSTD




