位置: IT常识 - 正文

【swinUnet官方代码测试自己的数据集(已训练完毕)】

编辑:rootadmin
【swinUnet官方代码测试自己的数据集(已训练完毕)】 ***************************************************码字不易,收藏之余,别忘了给我点个赞吧!***************************************************---------Start

推荐整理分享【swinUnet官方代码测试自己的数据集(已训练完毕)】,希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:,内容如对您有帮助,希望把文章链接给更多的朋友!

首先参考上一篇的训练过程,因为测试需要用到训练获得的权重。

1、检查相关文件1.1 检查test_vol.txt的内容是否是测试用的npz文件名称【swinUnet官方代码测试自己的数据集(已训练完毕)】

测试集的npz文件

1.2 检查模型权重文件

2、修改部分代码2.1 修改dataset_synapse.py

slice_name = self.sample_list[idx].strip('\n') data_path = os.path.join(self.data_dir, slice_name+'.npz') data = np.load(data_path) image, label = data['image'], data['label'] #改,numpy转tensor image = torch.from_numpy(image.astype(np.float32)) image = image.permute(2,0,1) label = torch.from_numpy(label.astype(np.float32))2.2 修改test.py代码

修改相关参数和文件路径 is_savenii:是否保存预测结果图片 num_classes:预测的目标类别数+1 cfg:swinUnet网络结构配置文件 test_save_dir:保存预测结果文件夹 num_classes:预测的目标类别数+1 自定义权重路径

2.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().detach().numpy() _, x, y = image.shape # 缩放图像符合网络输入大小224x224 if x != patch_size[0] or y != patch_size[1]: image = zoom(image, (1, patch_size[0] / x, patch_size[1] / y), order=3) input = torch.from_numpy(image).unsqueeze(0).float().cuda() net.eval() with torch.no_grad(): out = torch.argmax(torch.softmax(net(input), dim=1), dim=1).squeeze(0) out = out.cpu().detach().numpy() # 缩放预测结果图像同原始图像大小 if x != patch_size[0] or y != patch_size[1]: prediction = zoom(out, (x / patch_size[0], y / patch_size[1]), order=0) else: prediction = out metric_list = [] for i in range(1, classes): metric_list.append(calculate_metric_percase(prediction == i, label == i)) if test_save_path is not None: #保存预测结果 prediction = Image.fromarray(np.uint8(prediction)).convert('L') prediction.save(test_save_path + '/' + case + '.png') return metric_list第二种情况:保存可见图像,将不同类别映射成不同的颜色。只需要将上面代码的if test_save_path is not None:里面的内容替换成下面的代码即可。 #将不同类别区域呈彩色展示 #2分类 背景为黑色,类别1为绿色 if test_save_path is not None: a1 = copy.deepcopy(prediction) a2 = copy.deepcopy(prediction) a3 = copy.deepcopy(prediction) #r通道 a1[a1 == 1] = 0#g通道 a2[a2 == 1] = 255#b通道 a3[a3 == 1] = 0 a1 = Image.fromarray(np.uint8(a1)).convert('L') a2 = Image.fromarray(np.uint8(a2)).convert('L') a3 = Image.fromarray(np.uint8(a3)).convert('L') prediction = Image.merge('RGB', [a1, a2, a3]) prediction.save(test_save_path+'/'+case+'.png')

至此,设置完毕,右键run运行,若控制台出现下面的结果,则表示运行正确,我这里的权重只训练了一个epoch,所以预测的都是0。

3、查看预测结果

查看日志文件 查看预测结果图

总结: swinUnet主要由swin_transform模块构成,数据量太少的时候训练效果很差,跟TransUnet不能比。由于仅文字表述某些操作存在局限性,故只能简略描述,有任何疑问可下方留言评论或私信,回复不及还望见谅,感激不尽!
本文链接地址:https://www.jiuchutong.com/zhishi/293157.html 转载请保留说明!

上一篇:vue3使用svg图标多种方式(vue引用svg矢量图)

下一篇:Zend Framework教程之Zend_Registry对象用法分析(zendframework3)

  • 税收优惠属于政策吗
  • 小规模纳税人如何申报
  • 计提工会经费的科目
  • 制造业怎么核算成本
  • 金税四期什么时候执行
  • 所得税费用影响净利润吗
  • 分公司小规模
  • 苗木种植发票免增值税吗
  • 生产研发设备
  • 高新技术企业如何查询
  • 支付外汇货款
  • 审计核减理由
  • 平销返利企业所得税处理
  • 银行按揭方式销售开发产品
  • 企业取得5%的房产增值税发票是否可以抵扣呢?
  • 房地产开发结转成本条件
  • 驾校挂靠车辆账务处理?
  • 长期借款工程完工时会计处理怎么做?
  • 公司变卖后员工怎么办
  • 工伤抚恤金算遗产吗
  • 建安企业预缴个税怎么算
  • 年初未分配利润加本年净利润等于
  • 小规模纳税人附表一怎么填
  • 上海电商行业怎么样
  • 工商年检社保缴费基数按什么标准填
  • 没有把发票开给客户公司注销了怎么办
  • 居民企业之间以非货币资产进行投资
  • 商品返点收入账务处理
  • 外挂项目跨年结转分录怎么做?
  • 机器人销售属于什么行业
  • 预计负债什么时候确认
  • 鸿蒙超级终端搜不到
  • 主板故障开机断电
  • 企业退款如何走账
  • 赠与合同的法定撤销和任意撤销
  • linux中ping命令怎么用
  • 进项税大于销项税怎么结转
  • 企业汇算清缴怎么退税
  • 企业委托境外研发所发生的费用
  • 超支福利费按什么计提
  • 公司主营业务有哪些类型
  • 年末结转利润分配账户的借方余额表示
  • json对象和js对象
  • ros 2
  • 公司收到投资款怎么写收据?
  • 投资中间人要担什么责任
  • 固定资产加速折旧最新税收政策2023
  • 哪里能找到前端练手项目教程
  • 使用ajax实现页面分页
  • echo输出语句
  • fmt println
  • 代理记账费用计入什么二级科目
  • 账实不符的后果和对策
  • 社保费用如何做账,社保费会计分录如何写
  • 资产负债表日后至财务报告批准报出日之间
  • sql server 自定义排序
  • 安装购买的材料怎么做账
  • 应收账款属于会计分录
  • 支付利息的账务处理
  • 滞纳金在年报的营业外支出的哪一项?
  • 劳务派遣人员能有营业执照吗
  • 自定义winre
  • macos rootless
  • centos server
  • win8升级win10系统 激活
  • vm虚拟机中的浏览器在哪
  • linux AS3 oracle9i 安装指南
  • xp显示文件后缀怎么设置
  • linux gpfs
  • linux cp命令怎么用
  • windows8.
  • bootstrap 网格
  • dos命令怎么输入命令
  • unicode编码实现方案
  • [置顶]电影名字《收件人不详》
  • 南京税务局是地市级还是副省级
  • 进口汽车零部件编码查询
  • 增值税税控开票软件
  • 企业卖旧房如何计算缴纳土地增值税
  • 汽车装潢业务是干什么的
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设