红外弱小目标检测

红外弱小目标检测

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

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

在线体验

在这里插入图片描述

Link

如何开始

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

小目标数据集

数据集

  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值