Instructions to use zeyuren2002/EvalMDE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use zeyuren2002/EvalMDE with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("zeyuren2002/EvalMDE", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| #SBATCH --job-name=evalmde-smoke | |
| #SBATCH --output=/home/ywan0794/EvalMDE/smoke_evalmde_%j.log | |
| #SBATCH --error=/home/ywan0794/EvalMDE/smoke_evalmde_%j.log | |
| #SBATCH --open-mode=append | |
| #SBATCH --ntasks=1 | |
| #SBATCH --cpus-per-task=4 | |
| #SBATCH --gres=gpu:H100:1 | |
| #SBATCH --time=0-00:30:00 | |
| #SBATCH --mem=40G | |
| #SBATCH --nodelist=erinyes | |
| # Smoke test: Depth Pro inference on EvalMDE sample_data + sample_data_2, | |
| # then compute_metrics in evalmde env. End-to-end pipeline verification. | |
| set -u | |
| export PYTHONUNBUFFERED=1 | |
| cd /home/ywan0794/EvalMDE | |
| source /home/ywan0794/miniconda3/etc/profile.d/conda.sh | |
| DATA=/home/ywan0794/EvalMDE/data/smoke | |
| OUT=/home/ywan0794/EvalMDE/output/smoke | |
| mkdir -p $OUT | |
| echo "=== Smoke step 1: depth_pro inference ===" | |
| conda activate depth-pro | |
| export PYTHONPATH=/home/ywan0794/EvalMDE:/home/ywan0794/MoGe:${PYTHONPATH:-} | |
| python scripts/run_inference.py \ | |
| --baseline baselines/depth_pro.py \ | |
| --data-root $DATA \ | |
| --output-root $OUT \ | |
| --model-name depth_pro \ | |
| --repo /home/ywan0794/EvalMDE/ml-depth-pro \ | |
| --checkpoint /home/ywan0794/EvalMDE/ml-depth-pro/checkpoints/depth_pro.pt \ | |
| --precision fp32 | |
| echo "=== Smoke step 2: compute_metrics in evalmde env ===" | |
| conda deactivate; conda activate evalmde | |
| python scripts/compute_metrics.py \ | |
| --gt-root $DATA \ | |
| --pred-root $OUT \ | |
| --model-name depth_pro \ | |
| --output $OUT/depth_pro_metrics.json | |
| echo "=== Smoke summary ===" | |
| cat $OUT/depth_pro_metrics.json | |