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
| #ifndef TREE_H
| #define TREE_H
| #include "darknet.h"
|
| //typedef struct{
| // int *leaf;
| // int n;
| // int *parent;
| // int *child;
| // int *group;
| // char **name;
| //
| // int groups;
| // int *group_size;
| // int *group_offset;
| //} tree;
|
| #ifdef __cplusplus
| extern "C" {
| #endif
| //tree *read_tree(char *filename);
| int hierarchy_top_prediction(float *predictions, tree *hier, float thresh, int stride);
| void hierarchy_predictions(float *predictions, int n, tree *hier, int only_leaves);
| void change_leaves(tree *t, char *leaf_list);
| float get_hierarchy_probability(float *x, tree *hier, int c);
|
| #ifdef __cplusplus
| }
| #endif
| #endif
|
|
