mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
31 lines
641 B
C
31 lines
641 B
C
#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
|