Yoso Normal V1 8 1

🚀 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-8-1')
# 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
# Apply the model to the image
normal_image = predictor(input_image)
# Save or display the result
normal_image.save("output/normal_map.png")

📄 許可證

本項目採用Apache 2.0許可證。