Update utils.py
Browse files
utils.py
CHANGED
|
@@ -6,7 +6,7 @@ from torchvision.transforms import Compose, Resize, Grayscale, ToTensor, ToPILIm
|
|
| 6 |
|
| 7 |
# global variable for the grayscale transform
|
| 8 |
transform_gs = Compose(
|
| 9 |
-
[Resize((
|
| 10 |
)
|
| 11 |
|
| 12 |
def process_gs_image(image):
|
|
|
|
| 6 |
|
| 7 |
# global variable for the grayscale transform
|
| 8 |
transform_gs = Compose(
|
| 9 |
+
[Resize((360, 360)), Grayscale(num_output_channels=1), ToTensor()]
|
| 10 |
)
|
| 11 |
|
| 12 |
def process_gs_image(image):
|