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