This commit is contained in:
Joseph Redmon 2015-11-26 13:45:12 -08:00
parent 9b3da11a11
commit 064d4462d3
2 changed files with 2 additions and 1 deletions

View File

@ -10,7 +10,7 @@
#endif
char *voc_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"};
image voc_labels[80];
image voc_labels[20];
void train_yolo(char *cfgfile, char *weightfile)
{

View File

@ -21,6 +21,7 @@ extern "C" void convert_yolo_detections(float *predictions, int classes, int num
extern "C" void draw_yolo(image im, int num, float thresh, box *boxes, float **probs);
extern "C" char *voc_names[];
extern "C" image *voc_labels;
static float **probs;
static box *boxes;