TensorFlow.js中的通道优先格式

Question

我正在将https://github.com/Richard-An/StyleNAS转换为浏览器兼容的模型。我正在使用TensorFlow.js。由于转换库(https://github.com/nerox8664/pytorch2keras)中的一些问题，我希望将模型保留为channels_first (NCHW)格式。

我知道TensorFlow.js支持这一点。我可以成功地以所需格式导入我的模型。但我对数据有个问题。当我通过https://js.tensorflow.org/api/2.7.0/#browser.fromPixels将图像转换为张量时，它们始终是channels_last (NHWC)格式。有没有什么方法可以明确地改变这一点？我是TensorFlow.js的新手，对此我找不到任何答案。

谢谢

Answer 1

轴可以互换

HWC -> CHW (用于3d张量)

tensor.transpose([2, 0, 1])

NHWC -> NCHW (对于4d张量)

tensor.transpose([0, 3, 1, 2])