mat_utils.h
3.3 KB
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
// Tencent is pleased to support the open source community by making TNN available.
//
// Copyright (C) 2020 THL A29 Limited, a Tencent company. All rights reserved.
//
// Licensed under the BSD 3-Clause License (the "License"); you may not use this file except
// in compliance with the License. You may obtain a copy of the License at
//
// https://opensource.org/licenses/BSD-3-Clause
//
// Unless required by applicable law or agreed to in writing, software distributed
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
// specific language governing permissions and limitations under the License.
#ifndef TNN_INCLUDE_TNN_UTILS_MAT_UTILS_H_
#define TNN_INCLUDE_TNN_UTILS_MAT_UTILS_H_
#include "tnn/core/status.h"
#include "tnn/core/mat.h"
namespace TNN_NS {
typedef enum {
INTERP_TYPE_NEAREST = 0x00,
INTERP_TYPE_LINEAR = 0x01,
} PUBLIC InterpType;
typedef enum {
BORDER_TYPE_CONSTANT = 0x00,
BORDER_TYPE_REFLECT = 0x01,
BORDER_TYPE_EDGE = 0x02,
} PUBLIC BorderType;
typedef enum {
COLOR_CONVERT_NV12TOBGR = 0x00,
COLOR_CONVERT_NV12TOBGRA = 0x01,
COLOR_CONVERT_NV21TOBGR = 0x02,
COLOR_CONVERT_NV21TOBGRA = 0x03,
COLOR_CONVERT_BGRTOGRAY = 0x04,
COLOR_CONVERT_BGRATOGRAY = 0x05,
COLOR_CONVERT_RGBTOGRAY = 0x06,
COLOR_CONVERT_RGBATOGRAY = 0x07,
} PUBLIC ColorConversionType;
struct PUBLIC ResizeParam {
float scale_w = 0.0f;
float scale_h = 0.0f;
InterpType type = INTERP_TYPE_LINEAR;
};
struct PUBLIC CropParam {
int top_left_x = 0;
int top_left_y = 0;
int width = 0;
int height = 0;
};
struct PUBLIC WarpAffineParam {
float transform[2][3];
InterpType interp_type = INTERP_TYPE_NEAREST;
BorderType border_type = BORDER_TYPE_CONSTANT;
float border_val = 0.0f;
};
struct PUBLIC CopyMakeBorderParam {
int top = 0;
int bottom = 0;
int left = 0;
int right = 0;
BorderType border_type = BORDER_TYPE_CONSTANT;
float border_val = 0.0f;
};
class PUBLIC MatUtils {
public:
//copy cpu <-> device, cpu<->cpu, device<->device, src and dst dims must be equal.
static Status Copy(Mat& src, Mat& dst, void* command_queue);
//src and dst device type must be same. when param scale_w or scale_h is 0, it is computed as
// (double)dst.GetWidth() / src.GetWidth() or (double)dst.GetHeight() / src.GetHeight().
static Status Resize(Mat& src, Mat& dst, ResizeParam param, void* command_queue);
//src and dst device type must be same. when param width or height is 0, it is equal to
//dst.GetWidth() or dst.GetHeight().
static Status Crop(Mat& src, Mat& dst, CropParam param, void* command_queue);
//src and dst device type must be same.
static Status WarpAffine(Mat& src, Mat& dst, WarpAffineParam param, void* command_queue);
//src and dst device type must be same.
static Status CvtColor(Mat& src, Mat& dst, ColorConversionType type, void* command_queue);
//src and dst device type must be same. param top, bottom, left and right must be non-negative.
static Status CopyMakeBorder(Mat& src, Mat& dst, CopyMakeBorderParam param, void* command_queue);
};
} // namespace TNN_NS
#endif // TNN_INCLUDE_TNN_UTILS_MAT_UTILS_H_