mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed sequence of freeing memory.

Browse Source
This commit is contained in:
AlexeyAB
2017-03-28 13:57:27 +03:00
parent 5e5987baf8
commit 2662cd51a3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/yolo_v2_class.cpp
View File

@ -79,9 +79,9 @@ YOLODLL_API Detector::~Detector()
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);