Home IT技术 使用预训练主干网络的U-net:跳跃连接应放在哪里? 使用预训练主干网络的U-net:跳跃连接应放在哪里? IT技术 xiaolong · 2025年5月24日 · 0 Comment 我正在尝试使用PyTorch实现U-Net,并在编码器路径中使用预训练网络。 原始的U-Net论文是从头开始训练网络的。当使用预训练的主干网络时,有没有关于跳跃连接应该放置在哪里的资源或原则? 我已经找到了一些例子(例如这个仓库),但没有关于特征选择的任何理由说明。 回答: 在原始的U-Net论文中,跳跃连接使用的是Max-Pool层之前的特征。 使用预训练的主干网络时,逻辑完全相同:在每个空间分辨率上,选择最深的特征层。感谢github上的qubvel在一个issue中指出这一点。 相关文章: 理解PyTorch CNN通道 如何在PyTorch中初始化权重? PyTorch: Variable数据必须是张量 — 数据已是张量 如何在PyTorch中手动计算整个数据集的误差? 如何在pytorch中计算两个矩阵所有行之间的余弦相似度 访问训练过的自编码器的降维表示 神经网络能学会简单的插值吗? 如何在PyTorch中构建一个具有两个输入的网络 变量长度数据的训练 – EEG数据分类 pytorch传递架构类型使用argparse image-segmentation machine-learning neural-network pytorch