Merge pull request #1017 from jing-vision/master

Expose YOLODLL_API layer* get_network_layer(network* net, int i);
This commit is contained in:
Alexey
2018-06-10 14:48:38 +03:00
committed by GitHub
2 changed files with 6 additions and 0 deletions

View File

@ -493,6 +493,11 @@ image get_network_image_layer(network net, int i)
return def;
}
layer* get_network_layer(network* net, int i)
{
return net->layers + i;
}
image get_network_image(network net)
{
int i;

View File

@ -134,6 +134,7 @@ int resize_network(network *net, int w, int h);
void set_batch_network(network *net, int b);
int get_network_input_size(network net);
float get_network_cost(network net);
YOLODLL_API layer* get_network_layer(network* net, int i);
YOLODLL_API detection *get_network_boxes(network *net, int w, int h, float thresh, float hier, int *map, int relative, int *num, int letter);
YOLODLL_API detection *make_network_boxes(network *net, float thresh, int *num);
YOLODLL_API void free_detections(detection *dets, int n);