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

Partial fixed

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
Show Stats 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
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

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;
#ifdef OPENCV
args.threads = 3;
//args.threads = 3;
IplImage* img = NULL;
float max_img_loss = 5;
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->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);