Yoso Normal V1 5

🚀 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-5')

# 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

# 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ライセンスの下で提供されています。