Spaces:
Sleeping
Sleeping
votuongquan2004@gmail.com
commited on
Commit
•
eca3272
1
Parent(s):
1c5c8a7
update data.py
Browse files- utils/data.py +1 -1
utils/data.py
CHANGED
@@ -177,7 +177,7 @@ def preprocess(
|
|
177 |
inputs[0, :, :, :] = inputs[0, :, :, :] - inputs[0, :, 0, 0].mean(axis=0)
|
178 |
inputs[1, :, :, :] = inputs[1, :, :, :] - inputs[1, :, 0, 0].mean(axis=0)
|
179 |
|
180 |
-
return np.
|
181 |
|
182 |
|
183 |
def random_sample_np(data: np.ndarray, size: int) -> np.ndarray:
|
|
|
177 |
inputs[0, :, :, :] = inputs[0, :, :, :] - inputs[0, :, 0, 0].mean(axis=0)
|
178 |
inputs[1, :, :, :] = inputs[1, :, :, :] - inputs[1, :, 0, 0].mean(axis=0)
|
179 |
|
180 |
+
return inputs[np.newaxis, :].astype(np.float32)
|
181 |
|
182 |
|
183 |
def random_sample_np(data: np.ndarray, size: int) -> np.ndarray:
|