Spaces:
Runtime error
Runtime error
Update gradio_utils.py
Browse files- 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
|
| 75 |
-
std=[
|
| 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 |
]
|