File size: 1,114 Bytes
7667a87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import os
from einops import rearrange
import numpy as np
import torch
def path_splitall(path):
allparts = []
while 1:
parts = os.path.split(path)
if parts[0] == path: # sentinel for absolute paths
allparts.insert(0, parts[0])
break
elif parts[1] == path: # sentinel for relative paths
allparts.insert(0, parts[1])
break
else:
path = parts[0]
allparts.insert(0, parts[1])
return allparts
def change_layout(
data,
in_layout='NHWT', out_layout='NHWT',
ret_contiguous=False
):
in_layout = " ".join(in_layout.replace("C", "1"))
out_layout = " ".join(out_layout.replace("C", "1"))
data = rearrange(data, f"{in_layout} -> {out_layout}")
if ret_contiguous:
if isinstance(data, np.ndarray):
data = np.ascontiguousarray(data)
elif isinstance(data, torch.Tensor):
data = data.contiguous()
else:
raise ValueError
return data
if __name__ == "__main__":
print(path_splitall("/data/sevir_lr/data/CATALOG.csv")) |