diff --git a/src/coco.c b/src/coco.c index ed53cef4..66fddf54 100644 --- a/src/coco.c +++ b/src/coco.c @@ -7,6 +7,9 @@ #include "parser.h" #include "box.h" +#ifdef OPENCV +#include "opencv2/highgui/highgui_c.h" +#endif char *coco_classes[] = {"person","bicycle","car","motorcycle","airplane","bus","train","truck","boat","traffic light","fire hydrant","stop sign","parking meter","bench","bird","cat","dog","horse","sheep","cow","elephant","bear","zebra","giraffe","backpack","umbrella","handbag","tie","suitcase","frisbee","skis","snowboard","sports ball","kite","baseball bat","baseball glove","skateboard","surfboard","tennis racket","bottle","wine glass","cup","fork","knife","spoon","bowl","banana","apple","sandwich","orange","broccoli","carrot","hot dog","pizza","donut","cake","chair","couch","potted plant","bed","dining table","toilet","tv","laptop","mouse","remote","keyboard","cell phone","microwave","oven","toaster","sink","refrigerator","book","clock","vase","scissors","teddy bear","hair drier","toothbrush"}; diff --git a/src/darknet.c b/src/darknet.c index c03ba5b1..0b69f409 100644 --- a/src/darknet.c +++ b/src/darknet.c @@ -6,8 +6,9 @@ #include "utils.h" #include "cuda.h" -#define _GNU_SOURCE -#include +#ifdef OPENCV +#include "opencv2/highgui/highgui_c.h" +#endif extern void run_imagenet(int argc, char **argv); extern void run_detection(int argc, char **argv); diff --git a/src/detection.c b/src/detection.c index f5957010..c9dea3e8 100644 --- a/src/detection.c +++ b/src/detection.c @@ -5,6 +5,9 @@ #include "parser.h" #include "box.h" +#ifdef OPENCV +#include "opencv2/highgui/highgui_c.h" +#endif char *class_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"}; diff --git a/src/image.c b/src/image.c index 92c90665..d86f752a 100644 --- a/src/image.c +++ b/src/image.c @@ -3,6 +3,11 @@ #include #include +#ifdef OPENCV +#include "opencv2/highgui/highgui_c.h" +#include "opencv2/imgproc/imgproc_c.h" +#endif + #define STB_IMAGE_IMPLEMENTATION #include "stb_image.h" #define STB_IMAGE_WRITE_IMPLEMENTATION diff --git a/src/image.h b/src/image.h index e3800257..b635ff1f 100644 --- a/src/image.h +++ b/src/image.h @@ -1,18 +1,12 @@ #ifndef IMAGE_H #define IMAGE_H - #include #include #include #include #include -#ifdef OPENCV -#include "opencv2/highgui/highgui_c.h" -#include "opencv2/imgproc/imgproc_c.h" -#endif - typedef struct { int h; int w; diff --git a/src/imagenet.c b/src/imagenet.c index 499a8d3a..aeb7e690 100644 --- a/src/imagenet.c +++ b/src/imagenet.c @@ -2,6 +2,10 @@ #include "utils.h" #include "parser.h" +#ifdef OPENCV +#include "opencv2/highgui/highgui_c.h" +#endif + void train_imagenet(char *cfgfile, char *weightfile) { data_seed = time(0);