Last change
on this file since 454 was 453, checked in by coas-nagasima, 4 years ago |
ファイルを追加
|
-
Property svn:eol-style
set to
native
-
Property svn:mime-type
set to
text/x-chdr;charset=UTF-8
|
File size:
1.2 KB
|
Line | |
---|
1 | #ifndef _REGION_LAYER
|
---|
2 | #define _REGION_LAYER
|
---|
3 |
|
---|
4 | #include <stdint.h>
|
---|
5 | #include "kpu.h"
|
---|
6 |
|
---|
7 | typedef struct
|
---|
8 | {
|
---|
9 | uint32_t obj_number;
|
---|
10 | struct
|
---|
11 | {
|
---|
12 | uint32_t x1;
|
---|
13 | uint32_t y1;
|
---|
14 | uint32_t x2;
|
---|
15 | uint32_t y2;
|
---|
16 | uint32_t class_id;
|
---|
17 | float prob;
|
---|
18 | } obj[10];
|
---|
19 | } obj_info_t;
|
---|
20 |
|
---|
21 | typedef struct
|
---|
22 | {
|
---|
23 | float threshold;
|
---|
24 | float nms_value;
|
---|
25 | uint32_t coords;
|
---|
26 | uint32_t anchor_number;
|
---|
27 | float *anchor;
|
---|
28 | uint32_t image_width;
|
---|
29 | uint32_t image_height;
|
---|
30 | uint32_t classes;
|
---|
31 | uint32_t net_width;
|
---|
32 | uint32_t net_height;
|
---|
33 | uint32_t layer_width;
|
---|
34 | uint32_t layer_height;
|
---|
35 | uint32_t boxes_number;
|
---|
36 | uint32_t output_number;
|
---|
37 | void *boxes;
|
---|
38 | float *input;
|
---|
39 | float *output;
|
---|
40 | float *probs_buf;
|
---|
41 | float **probs;
|
---|
42 | } region_layer_t;
|
---|
43 |
|
---|
44 | typedef void (*callback_draw_box)(uint32_t x1, uint32_t y1, uint32_t x2, uint32_t y2, uint32_t class, float prob);
|
---|
45 |
|
---|
46 | int region_layer_init(region_layer_t *rl, int width, int height, int channels, int origin_width, int origin_height);
|
---|
47 | void region_layer_deinit(region_layer_t *rl);
|
---|
48 | void region_layer_run(region_layer_t *rl, obj_info_t *obj_info);
|
---|
49 | void region_layer_draw_boxes(region_layer_t *rl, callback_draw_box callback);
|
---|
50 |
|
---|
51 | #endif // _REGION_LAYER
|
---|
Note:
See
TracBrowser
for help on using the repository browser.