位置: 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)

  • soul陌生人可见和广场可见区别(soul陌生人可见是什么意思)

    soul陌生人可见和广场可见区别(soul陌生人可见是什么意思)

  • 怎么消除抖音视频上的水印(怎么消除抖音视频上的账号水印)

    怎么消除抖音视频上的水印(怎么消除抖音视频上的账号水印)

  • 路由器设置后有信号,怎么上不了网(路由器设置后有两个wifi)

    路由器设置后有信号,怎么上不了网(路由器设置后有两个wifi)

  • 苹果手机充电总是一响一响的(苹果手机充电总是反复断开)

    苹果手机充电总是一响一响的(苹果手机充电总是反复断开)

  • airpod pro怎么充电(airpodspro充电怎么冲)

    airpod pro怎么充电(airpodspro充电怎么冲)

  • 十进制数56转换成二进制数是(十进制56转换为16进制)

    十进制数56转换成二进制数是(十进制56转换为16进制)

  • boss账号被锁定了多久才能解封(boss账号被锁定能申诉回来吗)

    boss账号被锁定了多久才能解封(boss账号被锁定能申诉回来吗)

  • 骁龙845的手机有哪些(骁龙845手机有哪些便宜的)

    骁龙845的手机有哪些(骁龙845手机有哪些便宜的)

  • 抖音怎么评论表情包(抖音怎么评论表情包图片)

    抖音怎么评论表情包(抖音怎么评论表情包图片)

  • 华为手机无线投屏怎么电视显示不出来(华为手机无线投屏怎么全屏)

    华为手机无线投屏怎么电视显示不出来(华为手机无线投屏怎么全屏)

  • ctrl加什么是新建文档(ctrl加啥)

    ctrl加什么是新建文档(ctrl加啥)

  • 苹果x摄像头型号(苹果x摄像头长啥样)

    苹果x摄像头型号(苹果x摄像头长啥样)

  • 未接电话详单能查到吗(未接电话打电话记录详单能查到吗)

    未接电话详单能查到吗(未接电话打电话记录详单能查到吗)

  • 苹果怎么清理系统内存(苹果怎么清理系统文件)

    苹果怎么清理系统内存(苹果怎么清理系统文件)

  • 微信头像怎么加五星红旗(微信头像怎么加边框自己做)

    微信头像怎么加五星红旗(微信头像怎么加边框自己做)

  • 表示团结一致的谚语(表示团结一致的词语四字词语)

    表示团结一致的谚语(表示团结一致的词语四字词语)

  • 手机nnbs能删吗(手机上哪些可以删除)

    手机nnbs能删吗(手机上哪些可以删除)

  • 无需物流怎么修改(无需物流发货怎么办)

    无需物流怎么修改(无需物流发货怎么办)

  • 苹果退款要多久时间(苹果退款要多久到账)

    苹果退款要多久时间(苹果退款要多久到账)

  • 小米手环4nfc支持城市有哪些(小米手环4NFC支持比亚迪解锁吗)

    小米手环4nfc支持城市有哪些(小米手环4NFC支持比亚迪解锁吗)

  • qq扩列匹配为什么匹配不到(qq扩列还有匹配次数,为什么匹配失败)

    qq扩列匹配为什么匹配不到(qq扩列还有匹配次数,为什么匹配失败)

  • win10 enterprise是什么版本

    win10 enterprise是什么版本

  • 优酷弹幕怎么设置(优酷弹幕怎么设置不了)

    优酷弹幕怎么设置(优酷弹幕怎么设置不了)

  • nxtal10是什么型号

    nxtal10是什么型号

  • (Select)解决:Element-ui 中 Select 选择器下拉框样式及输入框样式的修改问题(背景色透明与悬停背景色变化与下拉框边距变化等操作)

    (Select)解决:Element-ui 中 Select 选择器下拉框样式及输入框样式的修改问题(背景色透明与悬停背景色变化与下拉框边距变化等操作)

  • ChatGPT会取代你的岗位吗?

    ChatGPT会取代你的岗位吗?

  • 什么公司需要缴纳印花税
  • 税务师都有什么科目
  • 企业所得税应纳税所得额怎么算
  • 回购股票为什么股价下跌
  • 超市小票能否入库
  • 残保金减免需要每月发放不低于本市最低工资标准是应发
  • 自建厂房的进项税额能一次性抵扣吗
  • 取得虚开发票所得税分录如何处理?
  • 记账凭证摘要的重要性
  • 转款账户错误退款的账务处理
  • 去年多计提的生育津贴
  • 小规模企业发票跨月可以作废吗?
  • 货物正常报损的分录
  • 什么情况下进项税额不得从销项税额中抵扣
  • 全年一次性奖金税收优惠政策
  • 文化事业建设税计算方法
  • 报销单大写金额填写格式1000
  • 没给顾客发票需要罚款吗
  • 亏损企业是否可以上成长板
  • 境外企业在境外提供劳务
  • 专家费怎么做账没发票
  • 短期借款利息的处理
  • 支付给外单位的押金是其他应收款吗
  • 增值税发票已经认证可以撤销吗?
  • windows hello用不了怎么办
  • win10怎么删除搜索
  • 分配利润给股东怎么做分录
  • 研发费用加计扣除优惠明细表
  • 开出去的发票没有进项发票应怎么核算成本?
  • win7为什么还有人用
  • ph是什么文件
  • PHP:mcrypt_enc_get_supported_key_sizes()的用法_Mcrypt函数
  • 给外部人员发奖状怎么说
  • 二手房印花税怎么算2020
  • 前端打印语句
  • 企业所得税核定征收方法
  • php支付功能
  • 微信支付开发包
  • win11更新22468
  • 商贸公司用什么名字最好
  • sendmailadmin
  • 公对公退款必须退至原账户的规定
  • 辞退福利为什么不计入产品成本
  • 普票加专票超过45万怎么交税
  • sql server2005怎么打开
  • sql如何进行事务操作
  • 豆腐是农产品还是工业产品
  • 法定公益金和法定保险金
  • 预收账款要预交税金吗
  • 预付账款在何时可以转出
  • 核定应税所得率征收
  • 应收帐款坏帐损失怎么算
  • 如何进行汇算清缴
  • 公司软件使用制度
  • 企业年报修改后没有公示怎么办
  • 如何冲减费用做账
  • 汇总帐凭证处理
  • mysql按时间查询
  • mysql查询语句大全讲解
  • sqlserver 触发器 if 后边没执行
  • solaris查找文件命令
  • windows 文件名太长无法删除
  • linux中sed详细用法
  • 安装程序不运行怎么回事
  • linux diff用法
  • 文件fwpkc1nt.sys
  • Linux系统安全性体现在哪些方面?
  • nb3是什么文件
  • cocos2dx官方教程
  • ExtJS TabPanel beforeremove beforeclose使用说明
  • nodejs可以直接操作数据库
  • vue全家桶包含哪些
  • 仿京东首页html5
  • exe取消隐藏属性
  • js选中div
  • 网上办税如何打印电子凭证
  • 福建税务局网上办税
  • 国家税务局河北省税务局
  • 汽车销售环节要做什么
  • 营销服务协议书范本
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设