|
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') |
|
|
|
data = [{'rgb': i, 'depth': None, 'intrinsic': None, 'filename': os.path.basename(i), 'folder': i.split('/')[-3]} for i in rgbs] |
|
return data |
