Pytorch-Medical-Segmentation 你好，我是用2d图像训练时遇到问题

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

Open NBIronman opened this issue 2 years ago • 4 comments

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

Jun 09 '22 07:06 NBIronman

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

Jun 09 '22 07:06 NBIronman

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

Aug 23 '22 06:08 NickWang1122

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

Feb 19 '23 00:02 MontaEllis

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

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

Oct 11 '23 08:10 pipixiaipip