Sijuade commited on
Commit
ea6f52d
·
1 Parent(s): af90824

Update gradio_utils.py

Browse files
Files changed (1) hide show
  1. gradio_utils.py +2 -2
gradio_utils.py CHANGED
@@ -71,8 +71,8 @@ def infer_transform(IMAGE_SIZE=config.INFERENCE_IMAGE_SIZE):
71
  A.PadIfNeeded(
72
  min_height=IMAGE_SIZE, min_width=IMAGE_SIZE, border_mode=cv2.BORDER_CONSTANT
73
  ),
74
- A.Normalize(mean=[0.45484068, 0.43406072, 0.40103856],
75
- std=[0.23936155, 0.23471538, 0.23876129],
76
  max_pixel_value=255,),
77
  ToTensorV2(),
78
  ]
 
71
  A.PadIfNeeded(
72
  min_height=IMAGE_SIZE, min_width=IMAGE_SIZE, border_mode=cv2.BORDER_CONSTANT
73
  ),
74
+ A.Normalize(mean=[0,0,0],
75
+ std=[1,1,1],
76
  max_pixel_value=255,),
77
  ToTensorV2(),
78
  ]