Added include/darknet.h

This commit is contained in:
AlexeyAB
2019-01-06 23:51:38 +03:00
parent c56931dd75
commit 3ff5084590
30 changed files with 1823 additions and 308 deletions

View File

@ -1,20 +1,21 @@
#ifndef TREE_H
#define TREE_H
#include "darknet.h"
typedef struct{
int *leaf;
int n;
int *parent;
int *child;
int *group;
char **name;
//typedef struct{
// int *leaf;
// int n;
// int *parent;
// int *child;
// int *group;
// char **name;
//
// int groups;
// int *group_size;
// int *group_offset;
//} tree;
int groups;
int *group_size;
int *group_offset;
} tree;
tree *read_tree(char *filename);
//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);