burn_depth 0.1.0

burn depth pro model inference
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

# burn_depth 🔥📐😎


[![test](https://github.com/mosure/burn_depth/workflows/test/badge.svg)](https://github.com/Mosure/burn_depth/actions?query=workflow%3Atest)
[![crates.io](https://img.shields.io/crates/v/burn_depth.svg)](https://crates.io/crates/burn_depth)


burn [depth pro](https://github.com/apple/ml-depth-pro) model inference


| input               | metric depth               |
|-----------------------|-----------------------|
| ![Alt text](./assets/image/test.jpg)    | ![Alt text](./docs/test_depth.png)    |


## usage


```rust
use burn_depth::model::depth_pro::DepthPro;

let model = DepthPro::<InferenceBackend>::load("assets/model/depth_pro.pt")?;
let depth = model.forward(input);
```


## setup


- download `depth_pro.pt` to `assets/model/`
- `cargo run --bin import`
- `cargo run --example inference`