用于Tensorflow-slim-ResNet V2预训练模型的RGB或BGR?

时间:2018-02-14 14:32:20

标签: tensorflow conv-neural-network resnet tensorflow-slim

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

那么有人确切知道Resens_V2预训练模型Tensorflow slim库的训练顺序(BGR或RGB)吗?它在文件中读到:

enter image description here

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

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

1 个答案:

答案 0 :(得分:0)

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

脚本中指定的颜色空间是RGB,请参见第206行。