highresnet的pytorch实现
highresnet的Python项目详细描述
highres3dnet
pytorch从Li et al., 2017, On the Compactness, Efficiency, and Representation of 3D Convolutional Networks: Brain Parcellation as a Pretext Task实现highres3dnet。
二维版本(HighRes2DNet
)也可用。
安装
PyTorch Hub
如果您使用的是夜间版本的pytorch,则可以导入模型 使用PyTorch Hub直接从这个存储库。
>>>importtorch>>>repo='fepegar/highresnet'>>>model_name='highres3dnet'>>>print(torch.hub.help(repo,model_name))"HighRes3DNet by Li et al. 2017 for T1-MRI brain parcellation""pretrained (bool): load parameters from pretrained model">>>model=torch.hub.load(repo,model_name,pretrained=True)
PyPI
$ pip install highresnet
>>>fromhighresnetimportHighRes3DNet>>>model=HighRes3DNet(in_channels=1,out_channels=160)