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