OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

NIPS demo

This commit is contained in:
Joseph Redmon 2015-06-10 00:22:48 -07:00
parent cbc9984a17
commit 2fb75da727
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Makefile
View File

@ -1,5 +1,5 @@
GPU=1
OPENCV=1
GPU=0
OPENCV=0
DEBUG=0
ARCH= -arch=sm_52
@ -42,7 +42,7 @@ endif
OBJS = $(addprefix $(OBJDIR), $(OBJ))
DEPS = $(wildcard src/*.h) Makefile
all: $(EXEC)
all: obj $(EXEC)
$(EXEC): $(OBJS)
$(CC) $(COMMON) $(CFLAGS) $(LDFLAGS) $^ -o $@
@ -53,6 +53,10 @@ $(OBJDIR)%.o: %.c $(DEPS)
$(OBJDIR)%.o: %.cu $(DEPS)
$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
obj:
mkdir -p obj
.PHONY: clean
clean:

9
src/detection.c
View File

@ -7,16 +7,16 @@
char *class_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"};
void draw_detection(image im, float *box, int side, int bg, char *label)
void draw_detection(image im, float *box, int side, char *label)
{
int classes = 20;
int elems = 4+classes+bg;
int elems = 4+classes;
int j;
int r, c;
for(r = 0; r < side; ++r){
for(c = 0; c < side; ++c){
j = (r*side + c) * elems + bg;
j = (r*side + c) * elems;
int class = max_index(box+j, classes);
if(box[j+class] > .2){
printf("%f %s\n", box[j+class], class_names[class]);
@ -212,6 +212,7 @@ void test_detection(char *cfgfile, char *weightfile)
load_weights(&net, weightfile);
}
detection_layer layer = get_network_detection_layer(net);
if (!layer.joint) fprintf(stderr, "Detection layer should use joint prediction to draw correctly.\n");
int im_size = 448;
set_batch_network(&net, 1);
srand(2222222);
@ -227,7 +228,7 @@ void test_detection(char *cfgfile, char *weightfile)
time=clock();
float *predictions = network_predict(net, X);
printf("%s: Predicted in %f seconds.\n", filename, sec(clock()-time));
draw_detection(im, predictions, 7, layer.background || layer.objectness, "predictions");
draw_detection(im, predictions, 7, "predictions");
free_image(im);
free_image(sized);
#ifdef OPENCV