Skip to content

Commit 1b94acc

Browse files
committed
fix the memory leaky bug for Yolov2.
1 parent ace1c00 commit 1b94acc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/caffe/data_transformer.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,7 @@ void DataTransformer<Dtype>::Transform_Yolo(const AnnotatedDatum& anno_datum,
448448
else {
449449
LOG(FATAL) << "Unknown annotation type.";
450450
}
451+
delete [] resized_image;
451452
}
452453

453454
template<typename Dtype>

0 commit comments

Comments
 (0)