File size: 689 Bytes
8d34f50 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import torch
import cv2
import numpy as np
import os
def load_dir(path, start, end):
lmss = []
for i in range(start, end):
datapath = os.path.join(path, str(i) + '.lms')
if os.path.isfile(datapath):
lms = np.loadtxt(os.path.join(
path, str(i) + '.lms'), dtype=np.float32)
lmss.append(lms)
#
# datapath = os.path.join(path, '{:d}.lms'.format(i))
# if os.path.isfile(datapath):
# lms = np.loadtxt(os.path.join(
# path, '{:d}.lms'.format(i)), dtype=np.float32)
# lmss.append(lms)
lmss = np.stack(lmss)
lmss = torch.as_tensor(lmss).cuda()
return lmss
|