基于图像分割的红外弱小目标检测算法
Infrared Target Detection by Segmentation (Deeplearing Method)
support Unet and FCN
Usage
Training
python main.py train
Test
python main.py test image_dir
Evaluation
python main.py evaluate
Visual DataSet
python main.py vis_dl
Datasets
- SIRST dataset is available at SIRST.
More Configs in main.py
TODO
- onnx export
- TensorRT Deploy
Model Weights
FCN weights
Unet weights