javascript - 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 的新手，我找不到任何答案。

谢谢

标签： javascripttensorflowcomputer-visiontensorflow.js

Answer 1

轴可以互换

HWC -> CHW（用于 3d 张量）

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

NHWC -> NCHW（用于 4d 张量）

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