tensorflow - Tensorflow-slim-ResNet V2 预训练模型的 RGB 或 BGR？

Question

对于 CNN 训练，输入图像 channel 的确切顺序可能因库而异，甚至因模型而异。对于 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 的训练顺序是什么？

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

Answer 1

它是RGB。色彩空间取决于在数据准备过程中图像是如何读入内存的。 Caffe 使用 OpenCV 进行许多图像操作，而 OpenCV 默认将图像读入 BGR，而在 TensorFlow 宇宙中，它更多地依赖于 PIL 库。

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