使用 RGB* 顺序训练的图像转换 TensorFlow* 模型的方法
- 使用原始 Keras 模型推断时,正确的结果。
- 将 Keras 模型转换为中间表示。
- 使用转换后的模型进行推断时,会产生错误的结果。
大多数 TensorFlow 模型都按 RGB 顺序训练图像。
默认情况下 , 推断引擎示例和演示期望以 BGR 通道顺序提供输入。
方法 1:
使用模型优化器工具进行模型转换,并--reverse_input_channels参数。
方法 2:
从 RGB 更改为 BGR:
im_cv = cv.imread('test2.jpg')
frame = cv.cvtColor(im_cv, cv.COLOR_RGB2BGR)
请参阅 何时反向输入通道