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

  • 房产税出现往期数据怎么办
  • 所得税的会计分录怎么做
  • 企业税收主要分为哪几类
  • 可以税前扣除的利息费用有哪些
  • 发票已抵扣但对方要红冲后续原发票要拿回来吗
  • 税率变更后账务怎么处理
  • 客票服务主要包括
  • 缴纳的增值税在资产负债表中怎么体现
  • 坏账核销计入营业利润吗
  • 分期付款方式购入货物会计分录
  • 打款金额少于开票金额
  • 银行本票结算分录
  • 出口退税的原则是什么
  • 转账支票给员工转工资
  • 小企业收到下列票据后,银行存款账户不变的是
  • 扣押属于纳税担保吗
  • 增值税一般纳税人税率
  • 教育附加费的计算方法
  • 一般纳税人注销库存需要补交税吗
  • 什么情况下税务会查账
  • 平销返利怎么开票入账
  • 报销单里报销内容可以写送客户烟吗
  • 进口材料怎么做账
  • 原材料结转生产成本是什么意思
  • 总账会计就是内外账合并吗
  • 出口货物怎么开具发票
  • 办劳务发票需要什么资料
  • 企业银行承兑汇票
  • linux的使用场合
  • 存续在业有什么区别
  • 企业对外股权投资涉及税收
  • 应交税金的含义
  • php判断ip是否属于某个区段
  • 报销没有发票如何处理做凭证
  • 税收筹划的风险性
  • 动态定位
  • el-table-column formatter
  • 外币借款业务的特点包括
  • 购买股票的佣金计入
  • 可供出售金融资产在资产负债表日反映的是
  • 分页的实现
  • php获取访问者mac地址
  • 最新版本金铲铲强势阵容
  • 社保缴费工资和实发工资不一样
  • node.js什么意思
  • js工厂模式创建对象
  • 以非现金资产清偿债务的,债权人应当
  • 事业单位用其他资金用于还单位贷款违法吗
  • 退伍军人企业所得税优惠
  • 公司房租收据怎么写
  • mongodb基础知识
  • 进口货物会计分录举例
  • 项目差旅费可以直接入成本吗
  • 公司收到银行发放贷款会计分录
  • 增值税一般纳税人是什么意思
  • 工程项目买保险没有合同 情况说明怎么写
  • 企业清算未分配利润怎么个税
  • 经营租赁不动产租赁发票开具要求
  • 公司有主营业务收入没有主营业务成本
  • win7系统出现reboot and
  • 服务器远程超出配置范围
  • bios如何设置
  • winoldap.exe - winoldap进程是什么意思
  • 苹果发布会亮点速览
  • wmiex.exe是什么程序
  • window sdk for windows 8.1
  • jquery实现
  • 怎么用winrar打包压缩文件
  • css隐藏显示元素的方法
  • dos替换文件命令
  • 深入解读2023年一号文件
  • exe取消隐藏属性
  • js实现瀑布流效果
  • 基于python的
  • python yield from 用法
  • javascript函数调用函数
  • 深入理解计算机系统
  • 公共基础设施项目目录
  • 公司自有房屋装修费用入什么料目
  • 个体户国税地税怎么交
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设