| | |
| | | for (int j = 0; j < FRAMES; ++j) free(detector_gpu.predictions[j]); |
| | | for (int j = 0; j < FRAMES; ++j) if(detector_gpu.images[j].data) free(detector_gpu.images[j].data); |
| | | |
| | | for (int j = 0; j < l.w*l.h*l.n; ++j) free(detector_gpu.probs[j]); |
| | | free(detector_gpu.boxes); |
| | | free(detector_gpu.probs); |
| | | for (int j = 0; j < l.w*l.h*l.n; ++j) free(detector_gpu.probs[j]); |
| | | |
| | | int old_gpu_index; |
| | | cudaGetDevice(&old_gpu_index); |
| | |
| | | cudaSetDevice(net.gpu_index); |
| | | //std::cout << "net.gpu_index = " << net.gpu_index << std::endl; |
| | | |
| | | float nms = .4; |
| | | //float nms = .4; |
| | | |
| | | image im; |
| | | im.c = img.c; |
| | |
| | | bbox.h = b.h*im.h; |
| | | bbox.obj_id = obj_id; |
| | | bbox.prob = prob; |
| | | bbox.track_id = 0; |
| | | |
| | | bbox_vec.push_back(bbox); |
| | | } |