clDNN
proposal.h
1 /*
2 // Copyright (c) 2017 Intel Corporation
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 
18 #ifndef PROPOSAL_H
19 #define PROPOSAL_H
20 
21 #include "cldnn.h"
28 
29 #ifdef __cplusplus
30 extern "C" {
31 #endif
32 
33 #define CLDNN_ROI_VECTOR_SIZE 5
34 
36  int max_proposals;
37  float iou_threshold;
38  int min_bbox_size;
39  int feature_stride;
40  int pre_nms_topn;
41  int post_nms_topn;
42  cldnn_float_arr ratios;
43  cldnn_float_arr scales;
45 
46 CLDNN_DECLARE_PRIMITIVE_TYPE_ID(proposal);
47 
48 #ifdef __cplusplus
49 }
50 #endif
51 
55 #endif /* PROPOSAL_H */
56 
#define CLDNN_BEGIN_PRIMITIVE_DESC(PType)
Begin primitive description definition.
Definition: cldnn.h:398
Represents reference to an array of floats.
Definition: cldnn.h:308
#define CLDNN_END_PRIMITIVE_DESC(PType)
Close primitive descriptor definition.
Definition: cldnn.h:409