TransUNet实现多波段影像语义分割

首先修改网络输入的通道数量 vit_seg_modeling_resnet_skip.py,将网络输入的通道数由3改成4。
在这里插入图片描述

修改数据加载代码 dataset_synapse.py,定义一个多波段影像读取函数,并将像素值inf替换为1,若不替换会导致后续训练出现损失为nan的情况。
在这里插入图片描述
最后修改预测代码 test.py,同样修改为能够加载多波段影像。
在这里插入图片描述

  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
【资源说明】 1.项目代码均经过功能验证ok,确保稳定可靠运行。欢迎下载使用体验! 2.主要针对各个计算机相关专业,包括计算机科学、信息安全、数据科学与大数据技术、人工智能、通信、物联网等领域的在校学生、专业教师、企业员工。 3.项目具有丰富的拓展空间,不仅可作为入门进阶,也可直接作为毕设、课程设计、大作业、初期项目立项演示等用途。 4.当然也鼓励大家基于此进行二次开发。在使用过程中,如有问题或建议,请及时沟通。 5.期待你能在项目中找到乐趣和灵感,也欢迎你的分享和反馈! 【项目介绍】 基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip 基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip 基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip 基于keras和Unet网络实现遥感图像语义分割源码+超详细注释(支持多波段+多类).zip

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值