Load normal map and transform it to normal bae system Adpated from https://github.com/modelscope/richdreamer/blob/main/dataset/gobjav

Load normal map and transform it to normal bae system Adpated from https://github.com/modelscope/richdreamer/blob/main/dataset/gobjav

Loads a glb model into the scene.

Loads a glb model into the scene.

load pipeline from hub or load from local ckpts: pipeline = IDArbDiffusionPipeline.from_pretrained("./pipeckpts")

Normalize OBJ file and return scale and offset.

Resets the scene to a clean state. Returns: None

将包含 normal 信息的张量绕 z 轴旋转 90 度 :param tensor: shape 为 (c, h, w)，其中 c = 3，表示 (x, y, z) 的 normal 信息 :return: 旋转后的张量

Saves rendered images of the object in the scene.

Saves rendered images of the object in the scene.

r""" Enable sliced attention computation. When this option is enabled, the attention module splits the input tensor in slices to comp

r""" Enable sliced attention computation. When this option is enabled, the attention module splits the input tensor in slices to comp

r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o

SSIM metric Args: numpy.ndarray

This is a warpper of the original cupy voronoi implementation The texture color where mask value is 1 will propagate to its n

r""" The call function to the pipeline for generation. Args: image (`PIL.Image.Image` or `List[PIL.Image.Image]` or `torc