Joseph Redmon
2015-11-26 064d4462d3890fd5dda5c7845119e1a3854d477b
stuff
2 files modified
3 ■■■■ changed files
src/yolo.c 2 ●●● patch | view | raw | blame | history
src/yolo_kernels.cu 1 ●●●● patch | view | raw | blame | history
src/yolo.c
@@ -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)
{
src/yolo_kernels.cu
@@ -21,6 +21,7 @@
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;