库名称: diffusers
流程标签: 图像到图像
许可证: apache-2.0

StableNormal模型说明文档

本代码库包含StableNormal的权重文件：通过降低扩散方差实现稳定锐利的法线贴图生成

使用方法

安装说明请参见GitHub仓库：https://github.com/Stable-X/StableNormal

模型使用示例如下：

import torch
from PIL import Image

# 加载输入图像
input_image = Image.open("路径/到/你的/图片.jpg")

# 创建预测器实例
predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-5')

# 使用alpha通道进行遮罩生成法线贴图
normal_map = predictor(rgba_image, data_type="object")  # 当存在alpha通道时将自动遮罩背景，否则使用birefnet
normal_map = predictor(rgba_image, data_type="outdoor")  # 使用Mask2Former遮罩天空和植物
normal_map = predictor(rgba_image, data_type="indoor") # 不进行遮罩处理

# 对输入图像应用模型
normal_image = predictor(input_image)

# 保存或显示结果
normal_image.save("output/法线贴图.png")