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

首先修改网络输入的通道数量 vit_seg_modeling_resnet_skip.py,将网络输入的通道数由3改成4。
TransUNet实现多波段影像语义分割_第1张图片

修改数据加载代码 dataset_synapse.py,定义一个多波段影像读取函数,并将像素值inf替换为1,若不替换会导致后续训练出现损失为nan的情况。
TransUNet实现多波段影像语义分割_第2张图片
最后修改预测代码 test.py,同样修改为能够加载多波段影像。
TransUNet实现多波段影像语义分割_第3张图片

你可能感兴趣的:(python,人工智能,深度学习)