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

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

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

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值