对于CNN培训,输入图像通道的确切顺序可能因库而异,甚至模型也不同。对于Caffe,输入图像通常预期为BGR顺序,而在Tensorflow中,顺序可以是任意的。
那么有人确切知道Resens_V2预训练模型Tensorflow slim库的训练顺序(BGR或RGB)吗?它在文件中读到:
我检查了此链接中的脚本:https://github.com/tensorflow/models/blob/master/research/slim/datasets/build_imagenet_data.py,它说图像是用RGB编码的。但是我还不确定ResNet_V2的训练顺序是什么?
有没有人对这个问题有类似的困惑?感谢您的任何反馈!
答案 0 :(得分:0)
它是RGB。颜色空间取决于在数据准备过程中如何将图像读取到内存中。 Caffe使用OpenCV进行许多图像操作,而OpenCV默认将图像读入BGR,而在TensorFlow Universe中,更常见的是依赖PIL库。
脚本中指定的颜色空间是RGB,请参见第206行。