Spaces:
Sleeping
Sleeping
Update preprocessing.py
Browse files- preprocessing.py +1 -4
preprocessing.py
CHANGED
@@ -4,7 +4,4 @@ import numpy as np
|
|
4 |
def pad_to_22_channels(input_tensor):
|
5 |
if input_tensor.shape[1] == 3: # RGB input
|
6 |
return torch.cat([input_tensor]*7 + [input_tensor[:,0:1]], dim=1)
|
7 |
-
return input_tensor
|
8 |
-
|
9 |
-
# Modify where you prepare inputs (usually near gmm() call)
|
10 |
-
inputA = pad_to_22_channels(your_rgb_input) # Before passing to GMM
|
|
|
4 |
def pad_to_22_channels(input_tensor):
|
5 |
if input_tensor.shape[1] == 3: # RGB input
|
6 |
return torch.cat([input_tensor]*7 + [input_tensor[:,0:1]], dim=1)
|
7 |
+
return input_tensor
|
|
|
|
|
|