def get_quat(self, im_index, cls_idx): cams = [] f_list = glob.glob(osp.join(self._model_path, self._classes[cls_idx], '*_camera.mat')) for f in f_list: cams.append(read_quat(f)) camq = np.stack(cams, axis=0) return camq
def get_quat(self, sid, mid, idx): rand_idx = idx cams = [] for ix in rand_idx: f = osp.join(self.im_dir, sid, mid, 'camera_{:d}.mat'.format(ix)) cams.append(read_quat(f)) camq = np.stack(cams, axis=0) return camq