| | |
| | | load_args args = {0}; |
| | | args.w = net.w; |
| | | args.h = net.h; |
| | | args.paths = paths; |
| | | args.c = net.c; |
| | | args.paths = paths; |
| | | args.n = imgs; |
| | | args.m = plist->size; |
| | | args.classes = classes; |
| | |
| | | args.hue = net.hue; |
| | | |
| | | #ifdef OPENCV |
| | | args.threads = 3; |
| | | args.threads = 3 * ngpus; |
| | | IplImage* img = NULL; |
| | | float max_img_loss = 5; |
| | | int number_of_lines = 100; |
| | |
| | | avg_loss = avg_loss*.9 + loss*.1; |
| | | |
| | | i = get_current_batch(net); |
| | | printf("\n %d: %f, %f avg, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), (what_time_is_it_now()-time), i*imgs); |
| | | printf("\n %d: %f, %f avg loss, %f rate, %lf seconds, %d images\n", get_current_batch(net), loss, avg_loss, get_current_rate(net), (what_time_is_it_now()-time), i*imgs); |
| | | |
| | | #ifdef OPENCV |
| | | if(!dont_show) |
| | |
| | | load_args args = { 0 }; |
| | | args.w = net.w; |
| | | args.h = net.h; |
| | | args.c = net.c; |
| | | args.type = IMAGE_DATA; |
| | | //args.type = LETTERBOX_DATA; |
| | | |
| | |
| | | |
| | | for (i = 0; i < m; ++i) { |
| | | char *path = paths[i]; |
| | | image orig = load_image_color(path, 0, 0); |
| | | image orig = load_image(path, 0, 0, net.c); |
| | | image sized = resize_image(orig, net.w, net.h); |
| | | char *id = basecfg(path); |
| | | network_predict(net, sized.data); |
| | |
| | | load_args args = { 0 }; |
| | | args.w = net.w; |
| | | args.h = net.h; |
| | | args.c = net.c; |
| | | args.type = IMAGE_DATA; |
| | | //args.type = LETTERBOX_DATA; |
| | | |
| | |
| | | { |
| | | list *options = read_data_cfg(datacfg); |
| | | char *name_list = option_find_str(options, "names", "data/names.list"); |
| | | char **names = get_labels(name_list); |
| | | int names_size = 0; |
| | | char **names = get_labels_custom(name_list, &names_size); //get_labels(name_list); |
| | | |
| | | image **alphabet = load_alphabet(); |
| | | network net = parse_network_cfg_custom(cfgfile, 1); // set batch=1 |
| | |
| | | } |
| | | //set_batch_network(&net, 1); |
| | | fuse_conv_batchnorm(net); |
| | | if (net.layers[net.n - 1].classes != names_size) { |
| | | printf(" Error: in the file %s number of names %d that isn't equal to classes=%d in the file %s \n", |
| | | name_list, names_size, net.layers[net.n - 1].classes, datacfg); |
| | | if(net.layers[net.n - 1].classes > names_size) getchar(); |
| | | } |
| | | srand(2222222); |
| | | double time; |
| | | char buff[256]; |
| | |
| | | if(!input) return; |
| | | strtok(input, "\n"); |
| | | } |
| | | image im = load_image_color(input,0,0); |
| | | image im = load_image(input,0,0,net.c); |
| | | int letterbox = 0; |
| | | //image sized = resize_image(im, net.w, net.h); |
| | | image sized = letterbox_image(im, net.w, net.h); letterbox = 1; |
| | | image sized = resize_image(im, net.w, net.h); |
| | | //image sized = letterbox_image(im, net.w, net.h); letterbox = 1; |
| | | layer l = net.layers[net.n-1]; |
| | | |
| | | //box *boxes = calloc(l.w*l.h*l.n, sizeof(box)); |