Beispiel #1
0
 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]))
Beispiel #2
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)