Marigold V2
Marigold V2 estimates depth, surface normals or intrinsic albedo with task-specific diffusion adapters.
- 任务
- depth, surface normals, albedo
- 尺寸
- b at 1024 px
- 安装
pip install "libreyolo[marigold]"- 支持层级
- 仅推理,自 v1.6.0 起。仅支持预测、验证和导出。训练相关的功能不适用。
- 许可
- 代码采用 MIT,权重采用 Apache-2.0。商用
Install
bash
pip install "libreyolo[marigold]"The default four-bit runtime requires CUDA.
Predict
Python
from libreyolo import LibreYOLO, SAMPLE_IMAGE # Requires CUDA for default four-bit inference.model = LibreYOLO("LibreMarigoldV2b-depth.pt", device="cuda")result = model(SAMPLE_IMAGE)print(result)The adapter downloads a separately pinned base model. Default quantization="4bit" requires CUDA; the snippet requires that runtime. quantization="none" selects the CPU-capable path. The default seed is 2025. Albedo results hold linear RGB and convert to sRGB for display. Training and export are not supported.
Validate
Python
from libreyolo import LibreYOLO, SAMPLE_IMAGE # Requires CUDA for default four-bit inference.model = LibreYOLO("LibreMarigoldV2b-depth.pt", device="cuda")# Depth dataset YAML: images/val plus same-stem maps in depths/val.metrics = model.val(data="path/to/your/depth.yaml", workers=0)print(metrics)Use the dataset format for this task. Validation explains the dataset requirements and returned metrics.
Checkpoints
| 文件 | 权重许可 |
|---|---|
| Depth | |
| LibreMarigoldV2b-depth.pt | apache-2.0 |
| LibreMarigoldV2b-depth-log-stage1.pt | apache-2.0 |
| LibreMarigoldV2b-depth-log-layered.pt | apache-2.0 |
| LibreMarigoldV2b-depth-uniform-base.pt | apache-2.0 |
| LibreMarigoldV2b-depth-uniform-layered.pt | apache-2.0 |
| LibreMarigoldV2b-depth-disparity-base.pt | apache-2.0 |
| LibreMarigoldV2b-depth-disparity-layered.pt | apache-2.0 |
| Surface normals | |
| LibreMarigoldV2b-normal.pt | apache-2.0 |
| Albedo | |
| LibreMarigoldV2b-albedo.pt | apache-2.0 |
上面的每个文件目前都在 LibreYOLO 组织中,并会在首次使用时下载。
Licensing
请检查你所下载的具体权重在 Hugging Face 仓库中的许可。LibreYOLO 组织里的每个检查点都附有许可,同一家族内也不一定相同。该仓库是权威来源;以下摘要说明本页上次验证时适用的情况。
这里只说明涉及的许可证,不构成法律意见。如果答案对商用很重要,请自行阅读许可证并咨询法律顾问。
- 原始工作
- Marigold V2, Huawei Bayer Lab
- 上游许可
- Apache-2.0
- LibreYOLO 代码
- MIT
- 权重
- 采用 Apache-2.0 许可,重新发布在 huggingface.co/LibreYOLO
- 解读
- The adapters and separately acquired Qwen image base declare Apache-2.0.