用于分割的动态增强

青葱年少 pytorch 216

原文标题 ：On the fly augmentation for segmentation

嗨，我正在尝试使用 U-net 进行 2D 前列腺分割。我想对前列腺和标签图像 (X,y) 应用数据增强，为此我目前正在使用 Torch.io，它将相同的随机变换应用于两个标签和数据本身。我的问题是我无法将所有增强数据加载到 RAM 中，因此想要进行即时增强。任何想法如何将转换后的数据直接馈送到我的 u-net 网络？？谢谢。

原文链接：https://stackoverflow.com//questions/71574218/on-the-fly-augmentation-for-segmentation

回复

我来回复
  • Shai的头像
    Shai 评论

    您应该查看诸如扩展 torchvision.transforms 以在分割掩码等上工作的数据库等库。

    2年前 0条评论

相关问题