File size: 1,192 Bytes
3ef1661 |
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 |
import glob
import os
import json
import cv2
def load_from_annos(anno_path):
with open(anno_path, 'r') as f:
annos = json.load(f)['files']
datas = []
for i, anno in enumerate(annos):
rgb = anno['rgb']
depth = anno['depth'] if 'depth' in anno else None
depth_scale = anno['depth_scale'] if 'depth_scale' in anno else 1.0
intrinsic = anno['cam_in'] if 'cam_in' in anno else None
normal = anno['normal'] if 'normal' in anno else None
data_i = {
'rgb': rgb,
'depth': depth,
'depth_scale': depth_scale,
'intrinsic': intrinsic,
'filename': os.path.basename(rgb),
'folder': rgb.split('/')[-3],
'normal': normal
}
datas.append(data_i)
return datas
def load_data(path: str):
rgbs = glob.glob(path + '/*.jpg') + glob.glob(path + '/*.png')
#intrinsic = [835.8179931640625, 835.8179931640625, 961.5419921875, 566.8090209960938] #[721.53769, 721.53769, 609.5593, 172.854]
data = [{'rgb': i, 'depth': None, 'intrinsic': None, 'filename': os.path.basename(i), 'folder': i.split('/')[-3]} for i in rgbs]
return data |