深度学习模型的数据读取(python+torch.utils.data.Dataset)
在用python编写CNN、UNet、VGG等模型时,需要先对数据集进行处理,即先读取image和label并进行预处理(比如设置图像大小等)。在这之前已经将BraTS2018的nii格式文件保存为png格式:https://blog.csdn.net/weixin_43330946/article/details/89576759下面的代码是我最近写UNet用于分割BraTS2018...
原创
2019-06-05 20:20:44 ·
3390 阅读 ·
3 评论