红外弱小目标检测

红外弱小目标检测

基于经典语义分割算法设计的弱小目标训练检测框架，支持Unet，FCN等模型，易二次拓展开发。

如果感觉对你有帮助的话，欢迎Start

如何开始

git clone https://github.com/Linaom1214/Infrared-detect-by-segmentation.git && cd Infrared-detect-by-segmentation.git 

训练

python main.py train

测试

python main.py test image_dir

指标计算

python main.py evaluate

可视化数据集

python main.py vis_dl

支持的数据集

SIRST dataset is available at 数据集链接

More Configs in main.py
TODO
onnx export
TensorRT Deploy

模型权重

FCN weights
Unet weights

我的更多相关信息

Github

版权声明：本文为CSDN博主「linaom」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_38870183/article/details/123007084