learn2splat / pycolmap /image.py
multimodalart's picture
multimodalart HF Staff
Learn2Splat ZeroGPU demo: app, vendored optgs, prebuilt sm_120 wheels
9196862 verified
Raw History Blame Contribute Delete
944 Bytes
# 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