search

MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
MIT License
857 stars 197 forks source link

你好，我是用2d图像训练时遇到问题 #46

Open NBIronman opened 2 years ago

NBIronman commented 2 years ago

在使用您的代码时，我设置2d图像训练，在main.py文件中255行和277行，应该怎么修改吗？

NBIronman commented 2 years ago

分别报RuntimeError: number of dims don't match in permute 以及ValueError: Input tensor must be 4D 这两个错误

NickWang1122 commented 2 years ago

你好,我也遇到了这个问题, model_output_one_hot = torch.nn.functional.one_hot(labels, num_classes=hp.out_class+1).permute(0,4,1,2,3)是这行代码报错 请问解决了吗？

MontaEllis commented 1 year ago

抱歉，我会在最近更新一下代码

pipixiaipip commented 1 year ago

在使用您的代码时，我设置2d图像训练，在main.py文件中255行和277行，应该怎么修改吗？

你好，请问您解决了吗？该如何解决啊