Joseph Redmon
2015-08-14 e625b32e060753e2029d52bc0150637f2a14cb72
Should fix for OpenCV 3.0
6 files modified
26 ■■■■■ changed files
src/coco.c 3 ●●●●● patch | view | raw | blame | history
src/darknet.c 5 ●●●●● patch | view | raw | blame | history
src/detection.c 3 ●●●●● patch | view | raw | blame | history
src/image.c 5 ●●●●● patch | view | raw | blame | history
src/image.h 6 ●●●●● patch | view | raw | blame | history
src/imagenet.c 4 ●●●● patch | view | raw | blame | history
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"};
src/darknet.c
@@ -6,8 +6,9 @@
#include "utils.h"
#include "cuda.h"
#define _GNU_SOURCE
#include <fenv.h>
#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);
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"};
src/image.c
@@ -3,6 +3,11 @@
#include <stdio.h>
#include <math.h>
#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
src/image.h
@@ -1,18 +1,12 @@
#ifndef IMAGE_H
#define IMAGE_H
#include <stdlib.h>
#include <stdio.h>
#include <float.h>
#include <string.h>
#include <math.h>
#ifdef OPENCV
#include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc_c.h"
#endif
typedef struct {
    int h;
    int w;
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);