blob: 85719a671b2db3f1b04a4d53a719312a6796e48f [file] [log] [blame]
Lun Dong3470dd02022-08-11 14:50:01 -07001/*
2 * Copyright 2022 Google LLC
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef SSD_POSTPROCESS_PIPELINE_H_
18#define SSD_POSTPROCESS_PIPELINE_H_
19
20#include <stdint.h>
21
22#include "ssd_postprocess/common.h"
23
24#ifdef __cplusplus
25extern "C" {
26#endif // __cplusplus
27
28// SSD post-processing pipeline
29void ssd_postprocess_pipeline(uint8_t** model_out, Boxes* boxes,
30 const int max_faces, const float iou_threshold);
31
32#ifdef __cplusplusS
33} // extern "C"
34#endif // __cplusplus
35
36#endif // SSD_POSTPROCESS_PIPELINE_H_