diff --git a/src/coco_kernels.cu b/src/coco_kernels.cu index 2ec09152..9ec0cb09 100644 --- a/src/coco_kernels.cu +++ b/src/coco_kernels.cu @@ -20,7 +20,7 @@ extern "C" image ipl_to_image(IplImage* src); extern "C" void convert_coco_detections(float *predictions, int classes, int num, int square, int side, int w, int h, float thresh, float **probs, box *boxes, int only_objectness); extern "C" char *coco_classes[]; -extern "C" image *coco_labels; +extern "C" image coco_labels[]; static float **probs; static box *boxes; diff --git a/src/yolo_kernels.cu b/src/yolo_kernels.cu index 26487a7e..e102f543 100644 --- a/src/yolo_kernels.cu +++ b/src/yolo_kernels.cu @@ -21,7 +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; +extern "C" image voc_labels[]; static float **probs; static box *boxes;