2

对于 CNN 训练,输入图像通道的确切顺序可能因库而异,甚至因模型而异。对于 Caffe,通常期望输入图像是 BGR 顺序,而在 Tensorflow 中,顺序可以是任意的。

那么有没有人知道 Tensorflow slim 库的 ResNet_V2 预训练模型是按什么顺序(BGR 或 RGB)训练的?它在文件中写道:

在此处输入图像描述

我检查了这个链接中的脚本:https ://github.com/tensorflow/models/blob/master/research/slim/datasets/build_imagenet_data.py ,它说图像是用RGB编码的。但是我仍然不确定 ResNet_V2 是按什么顺序训练的?

有没有人对这个问题有类似的困惑?感谢您的任何反馈!

4

1 回答 1

2

它是RGB。颜色空间取决于在数据准备期间图像是如何读入内存的。Caffe 使用 OpenCV 进行许多图像操作,OpenCV 默认将图像读取到 BGR 中,而在 TensorFlow 世界中更经常依赖 PIL 库。

脚本中声明的色彩空间是 RGB,见第 206 行。

于 2019-01-26T15:45:01.727 回答