用于分割的动态增强
您好，我正在尝试使用 U-net 进行 2D 前列腺分割。我想将数据增强应用于前列腺图像和标签图像 (X,y)，为此我目前使用 Torch.io，它将相同的随机转换应用于标签和数据本身。我的问题是我无法将所有增强数据加载到 RAM，因此想要进行即时增强。有什么想法可以直接将转换后的数据输入到我的 u-net 网络吗？？？谢谢。
Hi I am trying to do 2D prostate segmentation using U-net. I want to apply data augmentation to both prostate and label images (X,y), for this I am currently using Torch.io which applies the same random tranformation to both label and data itself. My problem is I cannot load all the augmented data to RAM thus want to do on-the-fly augmentation. Any ideas how can I directly feed the transformed data to my u-net network ???? Thanks.
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（1）
您应该查看诸如 albuminations 之类的库，它们扩展了
torchvision.transforms也可以在分割掩模等You should look at libraries such as albuminations that extend the
torchvision.transformsto work also on segmentation masks etc.