@@ -41,11 +41,10 @@ struct Roi
41
41
* @param[in] d_c channel for resized image
42
42
* @param[in] s_w width for input image
43
43
* @param[in] s_h height for input image
44
- * @param[in] s_c channel for input image
45
44
* @param[in] stream cuda stream
46
45
*/
47
46
extern void resize_bilinear_gpu (
48
- unsigned char * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h, int s_c,
47
+ unsigned char * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h,
49
48
cudaStream_t stream);
50
49
51
50
/* *
@@ -54,14 +53,12 @@ extern void resize_bilinear_gpu(
54
53
* @param[in] src image
55
54
* @param[in] d_w width for letterbox-ing
56
55
* @param[in] d_h height for letterbox-ing
57
- * @param[in] d_c channel for letterbox-ing
58
56
* @param[in] s_w width for input image
59
57
* @param[in] s_h height for input image
60
- * @param[in] s_c channel for input image
61
58
* @param[in] stream cuda stream
62
59
*/
63
60
extern void letterbox_gpu (
64
- unsigned char * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h, int s_c ,
61
+ unsigned char * dst, unsigned char * src, int d_w, int d_h, int s_w, int s_h,
65
62
cudaStream_t stream);
66
63
67
64
/* *
@@ -70,11 +67,10 @@ extern void letterbox_gpu(
70
67
* @param[in] src image
71
68
* @param[in] d_w width for a image
72
69
* @param[in] d_h height for a image
73
- * @param[in] d_c channel for a image
74
70
* @param[in] stream cuda stream
75
71
*/
76
72
extern void nchw_to_nhwc_gpu (
77
- unsigned char * dst, unsigned char * src, int d_w, int d_h, int d_c, cudaStream_t stream);
73
+ unsigned char * dst, unsigned char * src, int d_w, int d_h, cudaStream_t stream);
78
74
79
75
/* *
80
76
* @brief Unsigned char to float32 for inference
@@ -94,14 +90,12 @@ extern void to_float_gpu(
94
90
* @param[in] src image
95
91
* @param[in] d_w width for output
96
92
* @param[in] d_h height for output
97
- * @param[in] d_c channel for output
98
93
* @param[in] s_w width for input
99
94
* @param[in] s_h height for input
100
- * @param[in] s_c channel for input
101
95
* @param[in] stream cuda stream
102
96
*/
103
97
extern void resize_bilinear_letterbox_gpu (
104
- unsigned char * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h, int s_c ,
98
+ unsigned char * dst, unsigned char * src, int d_w, int d_h, int s_w, int s_h,
105
99
cudaStream_t stream);
106
100
107
101
/* *
@@ -111,15 +105,13 @@ extern void resize_bilinear_letterbox_gpu(
111
105
* @param[in] src image
112
106
* @param[in] d_w width for output
113
107
* @param[in] d_h height for output
114
- * @param[in] d_c channel for output
115
108
* @param[in] s_w width for input
116
109
* @param[in] s_h height for input
117
- * @param[in] s_c channel for input
118
110
* @param[in] norm normalization
119
111
* @param[in] stream cuda stream
120
112
*/
121
113
extern void resize_bilinear_letterbox_nhwc_to_nchw32_gpu (
122
- float * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h, int s_c ,
114
+ float * dst, unsigned char * src, int d_w, int d_h, int s_w, int s_h,
123
115
float norm, cudaStream_t stream);
124
116
125
117
/* *
@@ -129,17 +121,16 @@ extern void resize_bilinear_letterbox_nhwc_to_nchw32_gpu(
129
121
* @param[in] src image
130
122
* @param[in] d_w width for output
131
123
* @param[in] d_h height for output
132
- * @param[in] d_c channel for output
133
124
* @param[in] s_w width for input
134
125
* @param[in] s_h height for input
135
- * @param[in] s_c channel for input
136
126
* @param[in] batch batch size
137
- * @param[in] norm normalization
127
+ * @param[in] mean mean value for normalization
128
+ * @param[in] std standard deviation for normalization
138
129
* @param[in] stream cuda stream
139
130
*/
140
131
extern void resize_bilinear_letterbox_nhwc_to_nchw32_batch_gpu (
141
- float * dst, unsigned char * src, int d_w, int d_h, int d_c, int s_w, int s_h, int s_c , int batch,
142
- float norm , cudaStream_t stream);
132
+ float * dst, unsigned char * src, int d_w, int d_h, int s_w, int s_h, int batch,
133
+ const float * mean, const float * std , cudaStream_t stream);
143
134
144
135
/* *
145
136
* @brief Argmax on GPU
0 commit comments