diff --git a/main.py b/main.py index 4fda33f34d343799d8b585d5862e62c46ad64862..94f8adada7720e0e9bf0bd914f82ae3df8adbe86 100644 --- a/main.py +++ b/main.py @@ -74,8 +74,8 @@ def pre_process(img_, crop_size=513): resize_h, resize_w, _ = img_.shape # mean, std - image_mean = np.array(args.image_mean) - image_std = np.array(args.image_std) + image_mean = np.array(1.0) + image_std = np.array(0.5) img_ = (img_ - image_mean) / image_std # pad to crop_size