diff --git a/cnn_e2e/ofrecord_util.py b/cnn_e2e/ofrecord_util.py index 79bb629109584c5f394eb92dcc45776b0cec3964..56bd43897bea884e8c1866514b79376f6d2d425c 100644 --- a/cnn_e2e/ofrecord_util.py +++ b/cnn_e2e/ofrecord_util.py @@ -19,7 +19,7 @@ def load_imagenet(args, batch_size, data_dir, data_part_num, codec): shape=(args.image_size, args.image_size, 3), dtype=flow.float, codec=codec, - preprocessors=[flow.data.NormByChannelPreprocessor(args.rgb_mean, args.rgb_std)], + preprocessors=[flow.data.NormByChannelPreprocessor(args.rgb_mean[::-1], args.rgb_stdi[::-1])], ) label_blob_conf = flow.data.BlobConf(