本项目基于 Segment Anything Model (SAM) 对遥感影像进行定制化分割训练与推理，并配套了从数据预处理到模型微调、预测的完整流程。主要功能包括：
数据集预处理：可对大幅遥感影像进行裁剪与增强，并自动生成相应的点提示数据。
模型微调：在预训练的 SAM 模型基础上，通过冻结部分参数，定制化微调用于特定分割任务。
自定义损失函数：将 BCE 和 Dice Loss 相结合，从而提升分割精度和稳定性。
预测模块：基于微调后的模型对新图像进行分割预测，并计算相关性能指标。
以下将详细介绍各个模块的实现与使用方法。
数据集预处理：可对大幅遥感影像进行裁剪与增强，并自动生成相应的点提示数据。
模型微调：在预训练的 SAM 模型基础上，通过冻结部分参数，定制化微调用于特定分割任务。
自定义损失函数：将 BCE 和 Dice Loss 相结合，从而提升分割精度和稳定性。
预测模块：基于微调后的模型对新图像进行分割预测，并计算相关性能指标。
基于 Segment Anything Model (SAM) 对遥感影像进行定制化分割训练与推理，并配套了从数据预处理到模型微调、预测的完整流程。
声明：本文仅代表作者观点，不代表本站立场。如果侵犯到您的合法权益，请联系我们删除侵权资源！如果遇到资源链接失效，请您通过评论或工单的方式通知管理员。未经允许，不得转载，本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态
评论