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

NIPS demo

Browse Source
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 GPU=0
OPENCV=1 OPENCV=0
DEBUG=0 DEBUG=0
ARCH= -arch=sm_52 ARCH= -arch=sm_52
@ -42,7 +42,7 @@ endif
OBJS = $(addprefix $(OBJDIR), $(OBJ)) OBJS = $(addprefix $(OBJDIR), $(OBJ))
DEPS = $(wildcard src/*.h) Makefile DEPS = $(wildcard src/*.h) Makefile
all: $(EXEC) all: obj $(EXEC)
$(EXEC): $(OBJS) $(EXEC): $(OBJS)
$(CC) $(COMMON) $(CFLAGS) $(LDFLAGS) $^ -o $@ $(CC) $(COMMON) $(CFLAGS) $(LDFLAGS) $^ -o $@
@ -53,6 +53,10 @@ $(OBJDIR)%.o: %.c $(DEPS)
$(OBJDIR)%.o: %.cu $(DEPS) $(OBJDIR)%.o: %.cu $(DEPS)
$(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@ $(NVCC) $(ARCH) $(COMMON) --compiler-options "$(CFLAGS)" -c $< -o $@
obj:
mkdir -p obj
.PHONY: clean .PHONY: clean
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"}; 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 classes = 20;
int elems = 4+classes+bg; int elems = 4+classes;
int j; int j;
int r, c; int r, c;
for(r = 0; r < side; ++r){ for(r = 0; r < side; ++r){
for(c = 0; c < side; ++c){ for(c = 0; c < side; ++c){
j = (r*side + c) * elems + bg; j = (r*side + c) * elems;
int class = max_index(box+j, classes); int class = max_index(box+j, classes);
if(box[j+class] > .2){ if(box[j+class] > .2){
printf("%f %s\n", box[j+class], class_names[class]); 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); load_weights(&net, weightfile);
} }
detection_layer layer = get_network_detection_layer(net); 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; int im_size = 448;
set_batch_network(&net, 1); set_batch_network(&net, 1);
srand(2222222); srand(2222222);
@ -227,7 +228,7 @@ void test_detection(char *cfgfile, char *weightfile)
time=clock(); time=clock();
float *predictions = network_predict(net, X); float *predictions = network_predict(net, X);
printf("%s: Predicted in %f seconds.\n", filename, sec(clock()-time)); 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(im);
free_image(sized); free_image(sized);
#ifdef OPENCV #ifdef OPENCV