觉得较好的代码

Rusian_Stand

已于 2022-01-19 11:02:57 修改

阅读量524

点赞数

文章标签： python

于 2022-01-17 22:11:51 首次发布

本文链接：https://blog.csdn.net/Vpn_zc/article/details/122549645

版权

照片的读取：

   self.images_lis = sorted(glob(os.path.join(self.data_dir, 'image/*.png')))

或者：

       #好代码
        self.images_lis = [os.path.join(self.data_dir, f) for f in sorted(os.listdir(os.path.join(self.data_dir, 'image'))) \
                 if f.endswith('JPG') or f.endswith('jpg') or f.endswith('png')]

从四元数到旋转矩阵_loongkingwhat的博客-CSDN博客_四元数转旋转矩阵

四元数到旋转矩阵的公式。

def qvec2rotmat(qvec):
    return np.array([
        [1 - 2 * qvec[2]**2 - 2 * qvec[3]**2,
         2 * qvec[1] * qvec[2] - 2 * qvec[0] * qvec[3],
         2 * qvec[3] * qvec[1] + 2 * qvec[0] * qvec[2]],
        [2 * qvec[1] * qvec[2] + 2 * qvec[0] * qvec[3],
         1 - 2 * qvec[1]**2 - 2 * qvec[3]**2,
         2 * qvec[2] * qvec[3] - 2 * qvec[0] * qvec[1]],
        [2 * qvec[3] * qvec[1] - 2 * qvec[0] * qvec[2],
         2 * qvec[2] * qvec[3] + 2 * qvec[0] * qvec[1],
         1 - 2 * qvec[1]**2 - 2 * qvec[2]**2]])


def rotmat2qvec(R):
    Rxx, Ryx, Rzx, Rxy, Ryy, Rzy, Rxz, Ryz, Rzz = R.flat
    K = np.array([
        [Rxx - Ryy - Rzz, 0, 0, 0],
        [Ryx + Rxy, Ryy - Rxx - Rzz, 0, 0],
        [Rzx + Rxz, Rzy + Ryz, Rzz - Rxx - Ryy, 0],
        [Ryz - Rzy, Rzx - Rxz, Rxy - Ryx, Rxx + Ryy + Rzz]]) / 3.0
    eigvals, eigvecs = np.linalg.eigh(K)
    qvec = eigvecs[[3, 0, 1, 2], np.argmax(eigvals)]
    if qvec[0] < 0:
        qvec *= -1
    return qvec

从旋转矩阵到四元数_loongkingwhat的博客-CSDN博客_旋转矩阵转四元数