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
def read_hparams(path): kvs = pickle.loads(file_util.readbin(path)) hparams = tf.contrib.training.HParams(**kvs) return hparams
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
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