基于图像分割的红外弱小目标检测算法

Infrared Target Detection by Segmentation (Deeplearing Method)

support Unet and FCN

2022.5.15 增加onnx模型导出

新增在线预测Demo

Demo

Usage

Training

python main.py unet train

Test

python main.py unet test image_dir

Evaluation

python main.py unet evaluate

Visual DataSet

python main.py unet  vis_dl

Export Onnx

python main.py unet  export

Datasets

• SIRST dataset is available at SIRST.

More Configs in main.py

TODO

• TensorRT Deploy

Model Weights

FCN weights

Unet weights