红外弱小目标检测

文章目录[隐藏]

基于经典语义分割算法设计的弱小目标训练检测框架，支持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

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