提交 ff98547d 编写于 作者: S ShawnXuan

rgb mean/std -> bgr

上级 ee7c9221
...@@ -19,7 +19,7 @@ def load_imagenet(args, batch_size, data_dir, data_part_num, codec): ...@@ -19,7 +19,7 @@ def load_imagenet(args, batch_size, data_dir, data_part_num, codec):
shape=(args.image_size, args.image_size, 3), shape=(args.image_size, args.image_size, 3),
dtype=flow.float, dtype=flow.float,
codec=codec, 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( label_blob_conf = flow.data.BlobConf(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册