基于图像分割的红外弱小目标检测算法 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
