我是靠谱客的博主 犹豫小刺猬,这篇文章主要介绍SwinUnet官方代测试数据集,现在分享给大家,希望可以做个参考。

1、修改部分代码
1.1 修改dataset_synapse.py
在这里插入图片描述
1.2 修改test.py代码
修改相关参数和文件路径
is_savenii:是否保存预测结果图片
num_classes:预测的目标类别数+1
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
1.3 修改util.py代码(分两种情况)
第一种情况:保存预测原图,保存的结果是一张灰度图,每个像素的值代表该像素属于哪个类别。例如(0:背景,1:目标1,2:目标2…),这是一张全黑图。


def test_single_volume(image, label, net, classes, patch_size=[256, 256], test_save_path=None, case=None, z_spacing=1):
    image, label = image.squeeze(0).cpu().detach().numpy(), label.squeeze(0).cpu().detac

最后

以上就是犹豫小刺猬最近收集整理的关于SwinUnet官方代测试数据集的全部内容,更多相关SwinUnet官方代测试数据集内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(52)

评论列表共有 0 条评论

立即
投稿
返回
顶部