mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

some changes, fix opencv

Browse Source
This commit is contained in:
Joseph Redmon 2016-06-09 14:52:09 -07:00
parent 7520949d84
commit 8a6ba2fff3
5 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -1,5 +1,5 @@
GPU=1 GPU=1
CUDNN=0 CUDNN=1
OPENCV=1 OPENCV=1
DEBUG=0 DEBUG=0

4
src/coco_demo.c
View File

@ -10,8 +10,8 @@
#define FRAMES 1 #define FRAMES 1
#ifdef OPENCV #ifdef OPENCV
#include "opencv2/highgui/highgui.hpp" #include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc.hpp" #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); 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[]; extern char *coco_classes[];

1
src/network.c
View File

@ -253,6 +253,7 @@ void backward_network(network net, network_state state)
int i; int i;
float *original_input = state.input; float *original_input = state.input;
float *original_delta = state.delta; float *original_delta = state.delta;
state.workspace = net.workspace;
for(i = net.n-1; i >= 0; --i){ for(i = net.n-1; i >= 0; --i){
state.index = i; state.index = i;
if(i == 0){ if(i == 0){

2
src/rnn.c
View File

@ -199,7 +199,7 @@ void train_char_rnn(char *cfgfile, char *weightfile, char *filename, int clear,
} }
} }
if(i%100==0){ if(i%1000==0){
char buff[256]; char buff[256];
sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i); sprintf(buff, "%s/%s_%d.weights", backup_directory, base, i);
save_weights(net, buff); save_weights(net, buff);

4
src/yolo_demo.c
View File

@ -8,8 +8,8 @@
#include <sys/time.h> #include <sys/time.h>
#ifdef OPENCV #ifdef OPENCV
#include "opencv2/highgui/highgui.hpp" #include "opencv2/highgui/highgui_c.h"
#include "opencv2/imgproc/imgproc.hpp" #include "opencv2/imgproc/imgproc_c.h"
image ipl_to_image(IplImage* src); 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); 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);