我需要将模型从 RGB 扩展到 RGBA。我可以处理模型上的代码重写,但我不想从头开始重新训练整个模型,而是希望从 3 个通道权重 + 零开始。
有没有简单的方法可以将手电筒的 3 个通道权重保存更改为 4 个?
python rgba pytorch
python ×1
pytorch ×1
rgba ×1