pycolmap/image.py

# Author: True Price <jtprice at cs.unc.edu>

import numpy as np

#-------------------------------------------------------------------------------
#
# Image
#
#-------------------------------------------------------------------------------

class Image:
    def __init__(self, name_, camera_id_, q_, tvec_):
        self.name = name_
        self.camera_id = camera_id_
        self.q = q_
        self.tvec = tvec_

        self.points2D = np.empty((0, 2), dtype=np.float64)
        self.point3D_ids = np.empty((0,), dtype=np.uint64)

    #---------------------------------------------------------------------------

    def R(self):
        return self.q.ToR()

    #---------------------------------------------------------------------------

    def C(self):
        return -self.R().T.dot(self.tvec)

    #---------------------------------------------------------------------------

    @property
    def t(self):
        return self.tvec