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

Partial fixed

Browse Source
This commit is contained in:
AlexeyAB
2019-02-07 00:15:31 +03:00
parent 64b217aa86
commit 0543278a5b
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
build/darknet/x64/partial.cmd
View File

@ -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 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 darknet.exe partial cfg/yolov2-tiny-voc.cfg yolov2-tiny-voc.weights yolov2-tiny-voc.conv.13 13

2
src/classifier.c
View File

@ -110,7 +110,7 @@ void train_classifier(char *datacfg, char *cfgfile, char *weightfile, int *gpus,
args.type = CLASSIFICATION_DATA; args.type = CLASSIFICATION_DATA;
#ifdef OPENCV #ifdef OPENCV
args.threads = 3; //args.threads = 3;
IplImage* img = NULL; IplImage* img = NULL;
float max_img_loss = 5; float max_img_loss = 5;
int number_of_lines = 100; int number_of_lines = 100;

11
src/parser.c
View File

@ -671,10 +671,13 @@ void parse_net_options(list *options, network *net)
net->policy = get_policy(policy_s); net->policy = get_policy(policy_s);
net->burn_in = option_find_int_quiet(options, "burn_in", 0); net->burn_in = option_find_int_quiet(options, "burn_in", 0);
#ifdef CUDNN_HALF #ifdef CUDNN_HALF
int compute_capability = get_gpu_compute_capability(net->gpu_index); if (net->gpu_index >= 0) {
if(get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1; int compute_capability = get_gpu_compute_capability(net->gpu_index);
else net->cudnn_half = 0; if (get_gpu_compute_capability(net->gpu_index) >= 700) net->cudnn_half = 1;
fprintf(stderr, " compute_capability = %d, cudnn_half = %d \n", compute_capability, net->cudnn_half); 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 #endif
if(net->policy == STEP){ if(net->policy == STEP){
net->step = option_find_int(options, "step", 1); net->step = option_find_int(options, "step", 1);