change to imagenet mean/std

open-mmlab · Jul 23, 2020 · 67a0b80 · 67a0b80
1 parent bbf6d2b
commit 67a0b80
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/pytorch2onnx.py b/tools/pytorch2onnx.py
@@ -103,15 +103,15 @@ def parse_args():
         help='input image size')
     parser.add_argument(
         '--mean',
-        type=int,
+        type=float,
         nargs='+',
-        default=[0, 0, 0],
+        default=[123.675, 116.28, 103.53],
         help='mean value used for preprocess input data')
     parser.add_argument(
         '--std',
         type=int,
         nargs='+',
-        default=[1, 1, 1],
+        default=[58.395, 57.12, 57.375],
         help='variance value used for preprocess input data')
     args = parser.parse_args()
     return args