Yoso Normal V1 7

🚀 稳定法线（StableNormal）

稳定法线（StableNormal）模型可减少扩散方差，生成稳定且清晰的法线图

🚀 快速开始

稳定法线（StableNormal）模型可用于减少扩散方差，以生成稳定且清晰的法线图。本仓库包含了该模型的权重。

关于安装说明，请参考GitHub仓库：https://github.com/Stable-X/StableNormal。

💻 使用示例

基础用法

import torch
from PIL import Image
# Load an image
input_image = Image.open("path/to/your/image.jpg")
# Create predictor instance
predictor = torch.hub.load("hugoycj/StableNormal", "StableNormal_turbo", trust_repo=True, yoso_version='yoso-normal-v1-7')
# Apply the model to the image
normal_image = predictor(input_image)
# Save or display the result
normal_image.save("output/normal_map.png")

高级用法

# Generate normal map using alpha channel for masking
normal_map = predictor(rgba_image, data_type="object")  # Will mask out background, if alpha channel is avalible, else use birefnet
normal_map = predictor(rgba_image, data_type="outdoor")  # Will use Mask2Former to mask out sky and plants
normal_map = predictor(rgba_image, data_type="indoor") # Will not mask out

📄 许可证

本项目采用Apache 2.0许可证。