MNN/docs/cpp/ImageProcess.md

255 lines
5.1 KiB
Markdown
Raw Normal View History

2022-08-31 20:11:16 +08:00
# ImageProcess
```cpp
class ImageProcess
```
## 枚举类
### ImageFormat
```cpp
enum ImageFormat {
RGBA = 0,
RGB = 1,
BGR = 2,
GRAY = 3,
BGRA = 4,
YCrCb = 5,
YUV = 6,
HSV = 7,
XYZ = 8,
BGR555 = 9,
BGR565 = 10,
YUV_NV21 = 11,
YUV_NV12 = 12,
YUV_I420 = 13,
HSV_FULL = 14,
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `RGBA` | |
| 1 | `RGB` | |
| 2 | `BGR` | |
| 3 | `GRAY` | |
| 4 | `BGRA` | |
| 5 | `YCrCb` | |
| 6 | `YUV` | |
| 7 | `HSV` | |
| 8 | `XYZ` | |
| 9 | `BGR555` | |
| 10 | `BGR565` | |
| 11 | `YUV_NV21` | |
| 12 | `YUV_NV12` | |
| 13 | `YUV_I420` | |
| 14 | `HSV_FULL` | |
---
### Filter
```cpp
enum Filter {
NEAREST = 0,
BILINEAR = 1,
BICUBIC = 2,
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `NEAREST` | 最近点 |
| 1 | `BILINEAR` | 双线性的 |
| 2 | `BICUBIC` | 双三次的 |
---
### Wrap
```cpp
enum Wrap {
CLAMP_TO_EDGE = 0,
ZERO = 1,
REPEAT = 2
};
```
| value | name | 说明 |
|:------|:-----|:-----------------|
| 0 | `CLAMP_TO_EDGE` | 固定边缘的 |
| 1 | `ZERO` | 零 |
| 1 | `REPEAT` | 重复 |
## 成员函数
---
### create
```cpp
static ImageProcess* create(const Config& config, const Tensor* dstTensor = nullptr);
```
为给定张量创建给定配置的图像处理
参数:
- `config` 给定配置
- `dstTensor` 给定张量默认为nullptr
返回:图像处理器
---
### create
```cpp
static ImageProcess* create(const ImageFormat sourceFormat = RGBA, const ImageFormat destFormat = RGBA,
const float* means = nullptr, const int meanCount = 0, const float* normals = nullptr,
const int normalCount = 0, const Tensor* dstTensor = nullptr);
```
为给定张量创建给定配置的图像处理
参数:
- `sourceFormat` 源图像格式默认为RGBA
- `destFormat` 目的图像格式默认为RGBA
- `means` 给定方法默认为nullptr
- `meanCount` 给定方法数量默认为0
- `normals` 给定常量默认为nullptr
- `normalCount` 给定常量数量默认为0
- `dstTensor` 给定张量默认为nullptr
返回:图像处理器
---
### ~ImageProcess
析构函数
---
### destroy
```cpp
static void destroy(ImageProcess* imageProcess);
```
释放图像资源
参数:
- `imageProcess` 被释放的图像进程
返回:`void`
---
### matrix
```cpp
inline const Matrix& matrix() const {
return mTransform;
};
```
得到仿射变换矩阵
参数:无
返回:仿射变换矩阵
---
### setMatrix
```cpp
void setMatrix(const Matrix& matrix);
```
设置仿射变换矩阵
参数:
- `matrix` 源仿射变换矩阵
返回:`void`
---
### convert
```cpp
ErrorCode convert(const uint8_t* source, int iw, int ih, int stride, Tensor* dest);
```
将源数据转换为给定的张量
参数:
- `source` 源资源数据
- `iw` 源资源数据的宽度
- `ih` 源资源数据的高度
- `stride` 每行的元素数100宽RGB包含至少300个元素
- `dest` 目的张量
返回结果code
---
### convert
```cpp
ErrorCode convert(const uint8_t* source, int iw, int ih, int stride, void* dest, int ow, int oh, int outputBpp = 0,
int outputStride = 0, halide_type_t type = halide_type_of<float>());
```
将源数据转换为给定的张量
参数:
- `source` 源资源数据
- `iw` 源资源数据的宽度
- `ih` 源资源数据的高度
- `stride` 每行的元素数100宽RGB包含至少300个元素
- `dest` 目的张量
- `ow` 输出宽度
- `oh` 输出高度
- `outputBpp` 如果是0设置为保存和config.destFormat默认为0
- `outputStride` 如果为0设置为ow * outputBpp默认为0
2022-09-30 10:02:52 +08:00
- `type` 支持`halide_type_of<uint8_t>`和`halide_type_of<float>`,默认为`halide_type_of<float>`
返回结果code
---
### createImageTensor
```cpp
template <typename T>
static Tensor* createImageTensor(int w, int h, int bpp, void* p = nullptr) {
return createImageTensor(halide_type_of<T>(), w, h, bpp, p);
}
static Tensor* createImageTensor(halide_type_t type, int w, int h, int bpp, void* p = nullptr);
```
用给定的数据创建张量
参数:
- `type` 只支持halide_type_of和halide_type_of
- `w` 图像宽度
- `h` 图像高度
- `bpp` 每像素字节
- `p` 像素数据指针默认为nullptr
返回:创建的张量
---
### setPadding
```cpp
void setPadding(uint8_t value) {
mPaddingValue = value;
};
```
当wrap=ZERO时设置填充值
参数:
- `value` 填充值
返回:`void`
---
### setDraw
```cpp
void setDraw();
```
设置绘制模式
参数:无
返回:`void`
---
### draw
```cpp
void draw(uint8_t* img, int w, int h, int c, const int* regions, int num, const uint8_t* color);
```
绘制img区域的颜色
参数:
- `img` 要绘制的图像
- `w` 图像的宽度
- `h` 图像的高度
- `c` 图像的通道
- `regions` 要绘制的区域,大小为[num * 3]包含num x {y, xl, xr}
- `num` 区域数量
- `color` 要绘制的颜色
返回:`void`