【发布时间】:2018-04-17 10:57:28
【问题描述】:
我正在尝试实现一篇关于语义分割的论文,但我对如何对分割网络生成的预测图进行上采样以匹配输入图像大小感到困惑。
例如,我使用 Resnet101 的一个变体作为分割网络(论文中使用的)。使用这种网络结构,大小为 321x321 的输入(本文再次使用)会生成大小为 41x41xC 的最终预测图(C 是类数)。因为我必须进行像素级预测,所以我需要将其上采样到 321x321xC。 Pytorch 提供向上采样的功能,输出大小是预测图大小的倍数。所以,我这里不能直接使用那个方法。
因为这一步涉及到每个语义分割网络,我相信应该有一个标准的方法来实现它。
我将不胜感激任何指针。提前致谢。
【问题讨论】:
标签: neural-network computer-vision image-segmentation pytorch semantic-segmentation