关于数据的维度问题

lee-zq / 3DUNet-Pytorch

3DUNet implemented with pytorch

494 stars 110 forks source link

Open daisysong1 opened 2 years ago

daisysong1 commented 2 years ago

首先，非常感谢作者提供了开源代码！在处理数据的时候，有语句是将三维的数据用unsqueeze(0)增加了一个维度，不是很理解这么做的意义。希望有人能解答一下！！非常感谢！

yuncheng97 commented 2 years ago

增加一个channel维度，conv3d输入3D数据需要input的维度是5D的，就是（B，C，H，W，D）