scripts/convert_ndds_data.py
def load_renderings(data_dir, split):
  """Load images and metadata from disk."""
  f = 'transforms_{}.json'.format(split)
  with open(path.join(data_dir, f), 'r') as fp:
    meta = json.load(fp)
  images = []
  cams = []
  # raise()
  print('Loading imgs')
  for frame in meta['frames']:
    fname = os.path.join(data_dir, frame['file_path'] + '.png')
    with open(fname, 'rb') as imgin:
      image = np.array(Image.open(imgin), dtype=np.float32) / 255.
    cams.append(frame['transform_matrix'])
    images.append(image)
  ret = {}
  ret['images'] = np.stack(images, axis=0)
  print('Loaded all images, shape is', ret['images'].shape)
  ret['camtoworlds'] = np.stack(cams, axis=0)
  w = ret['images'].shape[2]
  camera_angle_x = float(meta['camera_angle_x'])
  print()
  ret['focal'] = .5 * w / np.tan(.5 * camera_angle_x)
  return ret
Load images and metadata from disk.
Load images and metadata from disk.
Load images and metadata from disk .
def load_renderings(data_dir, split):
  f = 'transforms_{}.json'.format(split)
  with open(path.join(data_dir, f), 'r') as fp:
    meta = json.load(fp)
  images = []
  cams = []
  print('Loading imgs')
  for frame in meta['frames']:
    fname = os.path.join(data_dir, frame['file_path'] + '.png')
    with open(fname, 'rb') as imgin:
      image = np.array(Image.open(imgin), dtype=np.float32) / 255.
    cams.append(frame['transform_matrix'])
    images.append(image)
  ret = {}
  ret['images'] = np.stack(images, axis=0)
  print('Loaded all images, shape is', ret['images'].shape)
  ret['camtoworlds'] = np.stack(cams, axis=0)
  w = ret['images'].shape[2]
  camera_angle_x = float(meta['camera_angle_x'])
  print()
  ret['focal'] = .5 * w / np.tan(.5 * camera_angle_x)
  return ret
def load_renderings ( data_dir , split ) : f = 'transforms_{}.json' . format ( split ) with open ( path . join ( data_dir , f ) , 'r' ) as fp : meta = json . load ( fp ) images = [ ] cams = [ ] print ( 'Loading imgs' ) for frame in meta [ 'frames' ] : fname = os . path . join ( data_dir , frame [ 'file_path' ] + '.png' ) with open ( fname , 'rb' ) as imgin : image = np . array ( Image . open ( imgin ) , dtype = np . float32 ) / 255. cams . append ( frame [ 'transform_matrix' ] ) images . append ( image ) ret = { } ret [ 'images' ] = np . stack ( images , axis = 0 ) print ( 'Loaded all images, shape is' , ret [ 'images' ] . shape ) ret [ 'camtoworlds' ] = np . stack ( cams , axis = 0 ) w = ret [ 'images' ] . shape [ 2 ] camera_angle_x = float ( meta [ 'camera_angle_x' ] ) print ( ) ret [ 'focal' ] = .5 * w / np . tan ( .5 * camera_angle_x ) return ret
Load images and metadata from disk.
Load images and metadata from disk .
"""Load images and metadata from disk.""" # raise()
data_dir
split