我曾经使用 keras，它遵循的图像格式是 [高度 x 宽度 x 通道 x 样本]。我决定改用 PyTorch。但我没有切换我的数据加载方案。所以现在我有形状为 HxWxCxS 的 numpy 数组，而不是 PyTorch 所需的 SxCxHxW。有没有人有任何想法来转换这个？
python opencv image keras pytorch
image ×1
keras ×1
opencv ×1
python ×1
pytorch ×1