def forward(self, vertices):
vertices = srf.projection(vertices, self.P, self.dist_coeffs,
self.orig_size)
return vertices
def forward(self, vertices, P=None, dist_coeffs=None):
P = self.P if P is None else P
dist_coeffs = self.dist_coeffs if dist_coeffs is None else dist_coeffs
vertices = srf.projection(vertices, P, dist_coeffs, self.orig_size)
return vertices
