位置: IT常识 - 正文

TransUnet官方代码测试自己的数据集(已训练完毕)(transit code)

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

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

文章相关热门搜索词:transit代码,transcode,transit代码,transitioncode,transformer代码讲解,transponder code,transcode,transit代码,内容如对您有帮助,希望把文章链接给更多的朋友!

首先参考上一篇的训练过程,这是测试过程,需要用到训练过程的权重。

1. TransUnet训练完毕之后,会生成权重文件(默认保存位置如下),snapshot_path为保存权重的路径。

权重文件

2. 修改test.py文件

调整数据集路径。 训练和测试时的图像设置相同大小,并设置主干模型的名称同训练时一致。

TransUnet官方代码测试自己的数据集(已训练完毕)(transit code)

配置数据集相关信息。 手动添加权重。

3. 设置DataLoader

设置DataLoader中参数num_workers=0。

4. 修改utils.py文件

替换utils.py中的test_single_volume函数,原网络输出的是0,1,2,3,4像素的图片,分别代表5个类别,直接显示均呈黑色。对此,我们通过像素调整,使每个类别呈现不同的颜色。

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 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: a1 = copy.deepcopy(prediction) a2 = copy.deepcopy(prediction) a3 = copy.deepcopy(prediction) a1[a1 == 1] = 255 a1[a1 == 2] = 0 a1[a1 == 3] = 255 a1[a1 == 4] = 20 a2[a2 == 1] = 255 a2[a2 == 2] = 255 a2[a2 == 3] = 0 a2[a2 == 4] = 10 a3[a3 == 1] = 255 a3[a3 == 2] = 77 a3[a3 == 3] = 0 a3[a3 == 4] = 120 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') return metric_list

**方便小伙伴理解这部分代码,特意做了个图,a1,a2,a3分别代表RGB三个通道,开始它们的值通过deepcopy函数直接赋值,故三者的值都是一样的。 这里拿类别1举例:a1[a12]=0代表R通道中输出结果为2的赋值0, a2[a22]=255代表G通道中输出结果为2的赋值255, a3[a3==2]=77代表B通道中输出结果为2的赋值77,(0,255,77)对应就是绿色,类别2就是绿色(轮子)。 然后通过Image.merge(‘RGB’, [a1, a2, a3])函数合并三个通道,此时prediction就成了三通道彩色图。

至此,设置完毕,右键run运行。

5. 测试结束

测试结束后,会在根目录下生成predictions文件夹,文件夹的内容如下。

本文链接地址:https://www.jiuchutong.com/zhishi/285082.html 转载请保留说明!

上一篇:【今日春节】 (© bingdian/iStock/Getty Images Plus)(今天春节是什么生肖)

下一篇:利尼扬蒂野生动物保护区中的一只雄性豹子,博茨瓦纳 (© Karine Aigner/Tandem Stills + Motion)

  • 房地产销售增值税纳税义务发生时间
  • 企业筹办期间发生的开办费会计分录
  • 汇算清缴交了税怎么做分录
  • 电子税务局中在哪看本月开票统计表
  • 交易性金融资产公允价值变动计入
  • 金蝶KIS专业版报表操作数非数值
  • 核定征收可以改查账征收吗
  • 小规模纳税人费用票怎么做账
  • 转让财产怎么交税
  • 收到借款确认书模板
  • 储备基金是用来干嘛的
  • 多层股权结构设计控制法
  • 购买软件能直接用吗
  • 应交增值税进项税额月底怎么处理
  • 小规模纳税人可以出口退税吗
  • 核定征收未分配利润分红分录
  • 销售苗木属于什么行业
  • 营改增的增值税税率
  • 6位开票代码是什么意思
  • 建筑企业无票支出如何处理
  • 采购与付款内部控制开题报告
  • 其他应收款借方和贷方什么意思
  • 入住酒店报销
  • 收购未税矿产品的其他单位,其纳税地点为收购地
  • 商用房的增值税是多少
  • 模具费属于什么费用
  • 微信收入与支出是怎么算的
  • win10家庭最新版本
  • 如何使用腾讯手机管家
  • 银行承兑汇票的转让一般通过什么渠道
  • 如何在数据透视表中增加一行
  • PHP:session_set_cookie_params()的用法_Session函数
  • 如何进入企业年报系统
  • iconfonf
  • 长期借款利息的会计科目
  • vue调用后端接口的方法
  • laravel ajax
  • 购入商品再卖出
  • 训练自己的GPT模型 中文改英文
  • Linux(CentOS)下PHP扩展PDO编译安装的方法
  • cicd框架
  • 一般纳税人开发票给小规模纳税人
  • 代扣代缴境外增值税时间
  • python 函数的返回值
  • 总公司所得税汇算
  • 本年缴纳上年的所得税填年报
  • 各人所得税法规定
  • 安装sql server 2008提示重新启动计算机
  • sql server更新数据
  • 进项留底是什么科目
  • 暂估入账的固定资产
  • 土地租赁费如何摊销
  • 应付职工薪酬中社保费怎么记账
  • 暂估成本的账务处理分录
  • 投资者投入固定资产成本
  • 持有至到期投资账务处理
  • 充值优惠怎么写
  • 什么情况下可以安排待岗
  • 明细账示例图
  • 残疾人保障金怎么申报
  • 营业收入是利润总额吗
  • 如何控制生产成本和利润
  • 提供劳务的收入计入什么科目
  • 约当产量法下的加权平均法怎么算
  • mac系统怎么打开任务管理器
  • fedora gnome
  • 系和系怎么区分
  • linux vi命令详解菜鸟教学
  • svcproc.exe - svcproc是什么进程 有什么作用
  • 电脑系统重装win8
  • centos 常用命令教程
  • pkjobs.exe - pkjobs是什么进程 有什么用
  • WIN10系统英文怎么改中文?
  • win81with update
  • python贪吃蛇游戏代码怎么运行
  • jquery写函数
  • 陕西宝鸡国税人才招聘
  • 在江西税务网交医保流程
  • 西乡国税局电话
  • 茶叶出口退税率为什么是9%不是13%
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设