Function decode

src/sample.py:51–78

(args, paths)

def decode(args, paths):
    from encoding.model import ShapeAutoEncoder
    from utils.triplane_util import load_triplane_data
    import glob

    # dist_util.setup_dist(args.gpu_id)
    
    log_dir = encoding_log_dir(args.tag)
    ae_model = ShapeAutoEncoder(log_dir, args)
    ae_model.load_ckpt("final")

    for path in paths:
        feat_maps = load_triplane_data(path, device=dist_util.dev(), compose=False)
        feat_maps = [fm.unsqueeze(0) for fm in feat_maps]

        save_dir = os.path.dirname(path)
        if args.vox:
            ae_model.decode_voxel(save_dir, feat_maps, args.reso)
        else:
            if args.copy_mtl:
                try:
                    mtl_path = glob.glob(os.path.join(os.path.dirname(args.data_path), "mesh/*.mtl"))[0]
                except:
                    mtl_path = None
            else:
                mtl_path = None
            ae_model.decode_texmesh(save_dir, feat_maps, args.reso, n_faces=args.n_faces, texture_reso=args.texreso,
                                    save_highres_mesh=False, n_surf_pc=-1, mtl_path=mtl_path, file_format=args.file_format)


if __name__ == "__main__":