def _load_metadata(self, filename, attr): if exists(filename): with open(filename, 'rb') as f: setattr(self, attr, pickle.load(f)) glog.info('{0}: {1}\tfrom {2}'.format( attr, exists(filename), file_utils.extract_basename(filename)[0]))
def file_lists_match(self, list2): list2 = [file_utils.extract_basename(f)[0] for f in list2] hash_table = dict.fromkeys(list2) all_included = True for i in self.frame_basenames: if i not in hash_table: all_included = False break return all_included
def read_annotation_from_filename(path_to_data, fname): basename, ext = file_utils.extract_basename(fname) annoname = join(path_to_data, 'anno', '{0}.yml'.format(basename)) with open(annoname) as data_file: data_yml = yaml.load(data_file) return data_yml
distance=100, zoom=135) vertices_field, indices_field = plane() field = gloo.Program(vertex_tex, fragment_tex) field.bind(vertices_field) field['position'] = vertices_field field['u_texture'] = data.get(join(opt.path_to_data, 'texture.png')) field['transform'] = trackball all_programs = [] for fid, fname in enumerate(filenames): (basename, ext) = file_utils.extract_basename(fname) print('Loading model {0}/{1}: {2}'.format(fid, len(filenames), basename)) path_to_pc = join(opt.path_to_data, 'scene3d') img = io.imread(join(path_to_pc, '{0}.jpg'.format(basename)), dtype=np.float32) vertices, indices = objload(join(path_to_pc, '{0}.obj'.format(basename))) vertices['texcoord'][:, 1] = 1.0 - vertices['texcoord'][:, 1] tex_program = gloo.Program(vertex_tex, fragment_tex) tex_program.bind(vertices) tex_program['u_texture'] = img tex_program['transform'] = trackball all_programs.append(tex_program)