| | |
| | | #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
|
| | | #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 |