https://software.intel.com/en-us/forums/computer-vision/topic/785538
“问题已经解决了。这是因为我使用的模型使用 channels_first 作为 GPU 训练的默认值,而 OPENVINO 需要 channels_last 用于 TF 模型。”
这些是什么意思?
我怎样才能改变它们?
我在网上找不到任何进一步的参考。
https://software.intel.com/en-us/forums/computer-vision/topic/785538
“问题已经解决了。这是因为我使用的模型使用 channels_first 作为 GPU 训练的默认值,而 OPENVINO 需要 channels_last 用于 TF 模型。”
这些是什么意思?
我怎样才能改变它们?
我在网上找不到任何进一步的参考。
通道首先意味着在特定张量(考虑照片)中,您将拥有(Number_Of_Channels, Height , Width)
.
通道最后意味着通道位于张量(n 维数组)中的最后一个位置。
例子:
(3,360,720) --- Channels first
(360,720,3) --- Channels last
其中 3 来自 RGB(彩色图像)。
默认情况下,TensorFlow 在配置中具有通道最后一个设置。
问题来自这样一个事实,即一些现在已经过时的框架(例如Theano
)采用渠道优先的方法;移植对于新手来说尤其是个问题。
您的问题的解决方案是以“Channels_Last”格式重新训练您的模型。
您可以使用带有模型优化器的 --disable_nhwc_to_nchw 将具有 NCHW 布局的 TF 模型转换为 IR。
NCHW - 通道第一
NHWC - 通道最后
N:batch_size, C:no.of.channels, H:input_img_height, W:input_img_width
默认情况下,MKLDNN-plugin 使用 NCHW 数据布局。