3

https://software.intel.com/en-us/forums/computer-vision/topic/785538

“问题已经解决了。这是因为我使用的模型使用 channels_first 作为 GPU 训练的默认值,而 OPENVINO 需要 channels_last 用于 TF 模型。”

这些是什么意思?

我怎样才能改变它们?

我在网上找不到任何进一步的参考。

4

3 回答 3

5

通道首先意味着在特定张量(考虑照片)中,您将拥有(Number_Of_Channels, Height , Width).

通道最后意味着通道位于张量(n 维数组)中的最后一个位置。

例子:

    (3,360,720) --- Channels first

    (360,720,3) --- Channels last

其中 3 来自 RGB(彩色图像)。

默认情况下,TensorFlow 在配置中具有通道最后一个设置。

问题来自这样一个事实,即一些现在已经过时的框架(例如Theano)采用渠道优先的方法;移植对于新手来说尤其是个问题。

您的问题的解决方案是以“Channels_Last”格式重新训练您的模型。

于 2019-06-25T13:05:32.357 回答
1

您可以使用带有模型优化器的 --disable_nhwc_to_nchw 将具有 NCHW 布局的 TF 模型转换为 IR。

于 2019-06-25T22:28:53.110 回答
0

NCHW - 通道第一
NHWC - 通道最后

N:batch_size, C:no.of.channels, H:input_img_height, W:input_img_width

默认情况下,MKLDNN-plugin 使用 NCHW 数据布局。

于 2020-01-21T08:16:09.190 回答