Go to the documentation of this file.00001
00002
00003
00004
00005 #ifndef SAN_OBJECT_DETECTOR_DETECTOR_H
00006 #define SAN_OBJECT_DETECTOR_DETECTOR_H
00007
00008 #ifdef OPENCV
00009
00010 #define __USE_C99_MATH
00011 #include <stdbool.h>
00012
00013 #include "network.h"
00014 #include "opencv2/core/types_c.h"
00015
00019 typedef struct darknet_object
00020 {
00021 char *label;
00022 float probability;
00023 unsigned short centroid_x;
00024 unsigned short centroid_y;
00025 unsigned short left_bot_x;
00026 unsigned short left_bot_y;
00027 unsigned short right_top_x;
00028 unsigned short right_top_y;
00029 } darknet_object;
00030
00034 bool darknet_detect(network *net, IplImage *ipl, float thresh,
00035 char **class_names, darknet_object **detected_objects, int
00036 *num_detected_objects);
00037
00041 network create_network(char *cfg_filename, char *weight_filename);
00042
00046 char **get_class_names(char *classnames_filename);
00047
00048 #endif
00049
00050 #endif //SAN_OBJECT_DETECTOR_DETECTOR_H