| | |
| | | float avg_loss = -1; |
| | | network *nets = calloc(ngpus, sizeof(network)); |
| | | |
| | | int iter_save; |
| | | iter_save = 100; |
| | | |
| | | srand(time(0)); |
| | | int seed = rand(); |
| | | int i; |
| | |
| | | |
| | | int init_w = net.w; |
| | | int init_h = net.h; |
| | | int iter_save; |
| | | iter_save = get_current_batch(net); |
| | | |
| | | load_args args = {0}; |
| | | args.w = net.w; |
| | |
| | | args.small_object = l.small_object; |
| | | args.d = &buffer; |
| | | args.type = DETECTION_DATA; |
| | | args.threads = 8; // 64 |
| | | args.threads = 64; // 8 |
| | | |
| | | args.angle = net.angle; |
| | | args.exposure = net.exposure; |
| | |
| | | |
| | | //if (i % 1000 == 0 || (i < 1000 && i % 100 == 0)) { |
| | | //if (i % 100 == 0) { |
| | | if(i >= iter_save) { |
| | | iter_save += 100; |
| | | if(i >= (iter_save + 100)) { |
| | | iter_save = i; |
| | | #ifdef GPU |
| | | if (ngpus != 1) sync_nets(nets, ngpus, 0); |
| | | #endif |
| | |
| | | } |
| | | image im = load_image_color(input,0,0); |
| | | image sized = resize_image(im, net.w, net.h); |
| | | //image sized = letterbox_image(im, net.w, net.h); |
| | | layer l = net.layers[net.n-1]; |
| | | |
| | | box *boxes = calloc(l.w*l.h*l.n, sizeof(box)); |