Joseph Redmon
2016-06-09 8a6ba2fff3ee1c14bca0aa0e0a909aba7057cc94
some changes, fix opencv
5 files modified
13 ■■■■ changed files
Makefile 2 ●●● patch | view | raw | blame | history
src/coco_demo.c 4 ●●●● patch | view | raw | blame | history
src/network.c 1 ●●●● patch | view | raw | blame | history
src/rnn.c 2 ●●● patch | view | raw | blame | history
src/yolo_demo.c 4 ●●●● patch | view | raw | blame | history
Makefile
@@ -1,5 +1,5 @@
GPU=1
CUDNN=0
CUDNN=1
OPENCV=1
DEBUG=0
src/coco_demo.c
@@ -10,8 +10,8 @@
#define FRAMES 1
#ifdef OPENCV
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc_c.h"
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 char *coco_classes[];
src/network.c
@@ -253,6 +253,7 @@
    int i;
    float *original_input = state.input;
    float *original_delta = state.delta;
    state.workspace = net.workspace;
    for(i = net.n-1; i >= 0; --i){
        state.index = i;
        if(i == 0){
src/rnn.c
@@ -199,7 +199,7 @@
            }
        }
        if(i%100==0){
        if(i%1000==0){
            char buff[256];
            sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
            save_weights(net, buff);
src/yolo_demo.c
@@ -8,8 +8,8 @@
#include <sys/time.h>
#ifdef OPENCV
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc_c.h"
image ipl_to_image(IplImage* src);
void convert_yolo_detections(float *predictions, int classes, int num, int square, int side, int w, int h, float thresh, float **probs, box *boxes, int only_objectness);