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