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

stuff

Browse Source
This commit is contained in:
Joseph Redmon 2015-06-11 11:04:48 -07:00
parent 9a63893f11
commit d41fbf638e
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Makefile
View File

@ -1,5 +1,5 @@
GPU=0 GPU=1
OPENCV=0 OPENCV=1
DEBUG=0 DEBUG=0
ARCH= -arch=sm_52 ARCH= -arch=sm_52

6
src/detection.c
View File

@ -18,7 +18,7 @@ void draw_detection(image im, float *box, int side, char *label)
for(c = 0; c < side; ++c){ for(c = 0; c < side; ++c){
j = (r*side + c) * elems; 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] > 0){
printf("%f %s\n", box[j+class], class_names[class]); printf("%f %s\n", box[j+class], class_names[class]);
float red = get_color(0,class,classes); float red = get_color(0,class,classes);
float green = get_color(1,class,classes); float green = get_color(1,class,classes);
@ -77,9 +77,9 @@ void train_detection(char *cfgfile, char *weightfile)
//plist = get_paths("/home/pjreddie/data/voc/no_2012_val.txt"); //plist = get_paths("/home/pjreddie/data/voc/no_2012_val.txt");
//plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt"); //plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt");
//plist = get_paths("/home/pjreddie/data/voc/val_2012.txt"); //plist = get_paths("/home/pjreddie/data/voc/val_2012.txt");
plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt"); //plist = get_paths("/home/pjreddie/data/voc/no_2007_test.txt");
//plist = get_paths("/home/pjreddie/data/coco/trainval.txt"); //plist = get_paths("/home/pjreddie/data/coco/trainval.txt");
//plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt"); plist = get_paths("/home/pjreddie/data/voc/all2007-2012.txt");
} }
paths = (char **)list_to_array(plist); paths = (char **)list_to_array(plist);
pthread_t load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer); pthread_t load_thread = load_data_detection_thread(imgs, paths, plist->size, classes, net.w, net.h, side, side, background, &buffer);

2
src/parser.c
View File

@ -167,7 +167,7 @@ detection_layer parse_detection(list *options, size_params params)
int rescore = option_find_int(options, "rescore", 0); int rescore = option_find_int(options, "rescore", 0);
int joint = option_find_int(options, "joint", 0); int joint = option_find_int(options, "joint", 0);
int objectness = option_find_int(options, "objectness", 0); int objectness = option_find_int(options, "objectness", 0);
int background = option_find_int(options, "background", 1); int background = option_find_int(options, "background", 0);
detection_layer layer = make_detection_layer(params.batch, params.inputs, classes, coords, joint, rescore, background, objectness); detection_layer layer = make_detection_layer(params.batch, params.inputs, classes, coords, joint, rescore, background, objectness);
option_unused(options); option_unused(options);
return layer; return layer;