影像组学学员成果分享:提取U-Net网络的深度学习特征

本文介绍了如何从U-Net网络中提取深度学习特征,包括在模型定义中添加特征返回、捕获特征、修改dataloader以获取图像名称,并将特征保存在features.txt中。这些特征可用于图像分类和分割任务,结合传统组学方法可进一步提升模型性能。文章还提到了不同验证方法和任务难度与论文水平的关系,以及相关服务的费用详情。
摘要由CSDN通过智能技术生成

影像组学路径图

深度学习特征提取

1.选定需要提取特征的layer, 在定义网络结构的py文件里(比如resnet.py)找到forward函数,将其表示为features后return出来,如下图所示(以U-Net为例):

我这里选择提取在U-Net中第三次下采样后再进行卷积的特征。由于特征的维度较大,为了方便查看输出,可以对特征进行maxpooling操作。

2.在forward函数里return出特征后,需要在train.py文件中将features接住。如下所示:

3.同时,为了查看输出特征所对应的name,需要对dataloader函数进行修改。即在__get__item()函数中return每个特征对应的image_name。如下图所示:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值