| | |
| | | int letter = 0; |
| | | int nboxes = 0; |
| | | detection *dets = get_network_boxes(&net, det.w, det.h, demo_thresh, demo_thresh, 0, 1, &nboxes, letter); |
| | | if (nms) do_nms_obj_v3(dets, nboxes, l.classes, nms); |
| | | if (nms) do_nms_obj(dets, nboxes, l.classes, nms); |
| | | |
| | | printf("\033[2J"); |
| | | printf("\033[1;1H"); |
| | |
| | | demo_classes = classes; |
| | | demo_thresh = thresh; |
| | | printf("Demo\n"); |
| | | net = parse_network_cfg_custom(cfgfile, 1); |
| | | net = parse_network_cfg_custom(cfgfile, 1); // set batch=1 |
| | | if(weightfile){ |
| | | load_weights(&net, weightfile); |
| | | } |
| | | set_batch_network(&net, 1); |
| | | //set_batch_network(&net, 1); |
| | | fuse_conv_batchnorm(net); |
| | | srand(2222222); |
| | | |