示例#1
0
def read_hparams_with_new_backwards_compatible_additions(path):
  """Reads hparams from a file and adds in any new backwards compatible ones."""
  kvs = pickle.loads(file_util.readbin(path))
  log.verbose('Loaded %s' % repr(kvs))
  if 'ident' not in kvs:
    ident = 'sif'  #  A default identifier for old checkpoints.
    log.info('Default ident!')
  else:
    ident = kvs['ident']
  new_additions = backwards_compatible_hparam_defaults(ident)
  for k, v in new_additions.items():
    if k not in kvs:
      log.verbose('Adding hparam %s:%s since it postdates the checkpoint.' %
                  (k, str(v)))
      kvs[k] = v
  # r50 is no longer supported, replace it:
  if 'cnna' in kvs and kvs['cnna'] == 'r50':
    kvs['cnna'] = 's50'
  hparams = tf.contrib.training.HParams(**kvs)

  return hparams
示例#2
0
def read_hparams(path):
  kvs = pickle.loads(file_util.readbin(path))
  hparams = tf.contrib.training.HParams(**kvs)
  return hparams
示例#3
0
 def gt_mesh_str(self):
   if not hasattr(self, '_gt_mesh_str'):
     self._gt_mesh_str = file_util.readbin(self.gt_path)
   return self._gt_mesh_str
示例#4
0
def read_pts_file(path):
  """Reads a .pts or a .sdf point samples file."""
  contents = file_util.readbin(path)
  points = np.frombuffer(contents, dtype=np.float32).reshape([-1, 4])
  return points