diff --git a/build/darknet/x64/partial.cmd b/build/darknet/x64/partial.cmd index acbe3fa9..2d9e022a 100644 --- a/build/darknet/x64/partial.cmd +++ b/build/darknet/x64/partial.cmd @@ -6,6 +6,9 @@ rem Download Yolo9000: http://pjreddie.com/media/files/yolo9000.weights rem darknet.exe partial cfg/tiny-yolo-voc.cfg tiny-yolo-voc.weights tiny-yolo-voc.conv.13 13 +darknet.exe partial cfg/darknet53_448.cfg darknet53_448.weights darknet53.conv.74 74 + + darknet.exe partial cfg/yolov2-tiny-voc.cfg yolov2-tiny-voc.weights yolov2-tiny-voc.conv.13 13 diff --git a/src/classifier.c b/src/classifier.c index f27d31d8..80175a2f 100644 --- a/src/classifier.c +++ b/src/classifier.c @@ -110,7 +110,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus, args.type = CLASSIFICATION_DATA; #ifdef OPENCV - args.threads = 3; + //args.threads = 3; IplImage* img = NULL; float max_img_loss = 5; int number_of_lines = 100; diff --git a/src/parser.c b/src/parser.c index 8d927f81..df671008 100644 --- a/src/parser.c +++ b/src/parser.c @@ -671,10 +671,13 @@ void parse_net_options(list *options, network *net) net->policy = get_policy(policy_s); net->burn_in = option_find_int_quiet(options, "burn_in", 0); #ifdef CUDNN_HALF - int compute_capability = get_gpu_compute_capability(net->gpu_index); - if(get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1; - else net->cudnn_half = 0; - fprintf(stderr, " compute_capability = %d, cudnn_half = %d \n", compute_capability, net->cudnn_half); + if (net->gpu_index >= 0) { + int compute_capability = get_gpu_compute_capability(net->gpu_index); + if (get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1; + else net->cudnn_half = 0; + fprintf(stderr, " compute_capability = %d, cudnn_half = %d \n", compute_capability, net->cudnn_half); + } + else fprintf(stderr, " GPU isn't used \n"); #endif if(net->policy == STEP){ net->step = option_find_int(options, "step", 1);