README.md 1.5 KB
Newer Older
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
## 模型概述
MiDas v2.1 large 单目深度估计模型

模型可通过输入图像估计其中的深度信息

模型权重转换自 [MiDas](https://github.com/intel-isl/MiDaS) 官方开源项目


## 模型安装

```shell
$hub install MiDaS_Large
```

## 效果展示
![效果展示](https://img-blog.csdnimg.cn/20201227112600975.jpg)

## API 说明

```python
def depth_estimation(
    images=None,
    paths=None,
    batch_size=1,
    output_dir='output',
    visualization=False
)
```

深度估计API

**参数**

* images (list\[numpy.ndarray\]): 图片数据,ndarray.shape 为 \[H, W, C\],默认为 None;
* paths (list\[str\]): 图片的路径,默认为 None;
* batch\_size (int): batch 的大小,默认设为 1;
* visualization (bool): 是否将识别结果保存为图片文件,默认设为 False;
* output\_dir (str): 图片的保存路径,默认设为 output。


**返回**

* res (list\[numpy.ndarray\]): 图像深度数据,ndarray.shape 为 \[H, W\]


## 预测代码示例

```python
import cv2
import paddlehub as hub

# 模型加载
# use_gpu:是否使用GPU进行预测
model = hub.Module(name='MiDaS_Large', use_gpu=False)

# 模型预测
result = model.depth_estimation(images=[cv2.imread('/PATH/TO/IMAGE')])

# or
# result = model.style_transfer(paths=['/PATH/TO/IMAGE'])
```

## 模型相关信息

### 模型代码

https://github.com/intel-isl/MiDaS

### 依赖

paddlepaddle >= 2.0.0rc0

paddlehub >= 2.0.0b1