#include "network.h"
#include "region_layer.h"
#include "cost_layer.h"
#include "utils.h"
#include "parser.h"
#include "box.h"
#include "demo.h"
#include "option_list.h"
#include "detector.h"
Go to the source code of this file.
Functions | |
static int | get_coco_image_id (char *filename) |
static void | print_cocos (FILE *fp, char *image_path, box *boxes, float **probs, int num_boxes, int classes, int w, int h) |
void | print_detector_detections (FILE **fps, char *id, box *boxes, float **probs, int total, int classes, int w, int h) |
void | print_imagenet_detections (FILE *fp, int id, box *boxes, float **probs, int total, int classes, int w, int h, int *map) |
void | run_detector (int argc, char **argv) |
void | test_detector (char *datacfg, char *cfgfile, char *weightfile, char *filename, float thresh) |
void | train_detector (char *datacfg, char *cfgfile, char *weightfile, int *gpus, int ngpus, int clear) |
void | validate_detector (char *datacfg, char *cfgfile, char *weightfile) |
void | validate_detector_recall (char *cfgfile, char *weightfile) |
Variables | |
static int | coco_ids [] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90} |
static int get_coco_image_id | ( | char * | filename | ) | [static] |
Definition at line 280 of file detector.c.
static void print_cocos | ( | FILE * | fp, |
char * | image_path, | ||
box * | boxes, | ||
float ** | probs, | ||
int | num_boxes, | ||
int | classes, | ||
int | w, | ||
int | h | ||
) | [static] |
Definition at line 286 of file detector.c.
void print_detector_detections | ( | FILE ** | fps, |
char * | id, | ||
box * | boxes, | ||
float ** | probs, | ||
int | total, | ||
int | classes, | ||
int | w, | ||
int | h | ||
) |
Definition at line 312 of file detector.c.
void print_imagenet_detections | ( | FILE * | fp, |
int | id, | ||
box * | boxes, | ||
float ** | probs, | ||
int | total, | ||
int | classes, | ||
int | w, | ||
int | h, | ||
int * | map | ||
) |
Definition at line 333 of file detector.c.
void run_detector | ( | int | argc, |
char ** | argv | ||
) |
Definition at line 640 of file detector.c.
void test_detector | ( | char * | datacfg, |
char * | cfgfile, | ||
char * | weightfile, | ||
char * | filename, | ||
float | thresh | ||
) |
Definition at line 558 of file detector.c.
void train_detector | ( | char * | datacfg, |
char * | cfgfile, | ||
char * | weightfile, | ||
int * | gpus, | ||
int | ngpus, | ||
int | clear | ||
) |
Definition at line 137 of file detector.c.
void validate_detector | ( | char * | datacfg, |
char * | cfgfile, | ||
char * | weightfile | ||
) |
Definition at line 356 of file detector.c.
void validate_detector_recall | ( | char * | cfgfile, |
char * | weightfile | ||
) |
Definition at line 480 of file detector.c.
int coco_ids[] = {1,2,3,4,5,6,7,8,9,10,11,13,14,15,16,17,18,19,20,21,22,23,24,25,27,28,31,32,33,34,35,36,37,38,39,40,41,42,43,44,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,67,70,72,73,74,75,76,77,78,79,80,81,82,84,85,86,87,88,89,90} [static] |
Definition at line 14 of file detector.c.