Ssd.pytorch: RGB vs BGR?

Created on 28 Jul 2017  路  4Comments  路  Source: amdegroot/ssd.pytorch

Hello,
I was looking at your implementation and I believe the input to your model is an image with RGB ordering. I was also looking at the keras implementation and they use BGR values. I have been also testing with an map evaluation script and it seems that I get better results ;using the weights that you provided from the original caffe implementation, when I use BGR instead of RBG. Do you happen to know which order should we follow when using the original caffe weights?

Thank you very much :)

Most helpful comment

It should be BGR, but currently the pretrained model was trained with RGB code so when using it an RGB input is expected, I am planning to contribute a pretrained model that was trained with BGR, then the inconsistency could be removed

All 4 comments

It should be BGR, but currently the pretrained model was trained with RGB code so when using it an RGB input is expected, I am planning to contribute a pretrained model that was trained with BGR, then the inconsistency could be removed

Thank you @alexkoltun :)

@alexkoltun how do you train with rgb when the vgg_fc_reduced weights from caffe is in bgr? Do you have a way to convert that to rgb or do you train the vgg_fc_reduced from scratch as well?

Same question. Is the pretrained model (vgg16_fc_reducedfc.pth) in BGR or RGB order?

Was this page helpful?
0 / 5 - 0 ratings

Related issues

LAWSSSS picture LAWSSSS  路  4Comments

sxyxf66 picture sxyxf66  路  4Comments

chienyiwang picture chienyiwang  路  3Comments

JingpengSun picture JingpengSun  路  4Comments

luhc15 picture luhc15  路  6Comments