About the way you read image

[haoweiz23]

In 'main_imagenet.py' , line 321:

x = Variable(transform(Image.fromarray(img))).cuda()

You use Image.fromarray(img) to load a array type image object. However, channel order in img is BGR (opencv) while the channel order in Image is RGB, which means this operation will confuse the channel order.