mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-05 22:19:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #9229 from alalek:dnn_torch_memory_leaks

Browse Source
This commit is contained in:
Alexander Alekhin 2017-07-25 13:23:18 +00:00
commit bcff9fd067
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/dnn/src/torch/torch_importer.cpp
View File

@ -100,7 +100,7 @@ struct TorchImporter : public ::cv::dnn::Importer
typedef std::map<String, std::pair<int, Mat> > TensorsMap;
Net net;
THFile *file;
cv::Ptr<THFile> file;
std::set<int> readedIndexes;
std::map<int, Mat> storages;
std::map<int, Mat> tensors;
@ -126,7 +126,7 @@ struct TorchImporter : public ::cv::dnn::Importer
rootModule = curModule = NULL;
moduleCounter = 0;
file = THDiskFile_new(filename.c_str(), "r", 0);
file = cv::Ptr<THFile>(THDiskFile_new(filename.c_str(), "r", 0), THFile_free);
CV_Assert(file && THFile_isOpened(file));
if (isBinary)
@ -976,18 +976,20 @@ struct TorchImporter : public ::cv::dnn::Importer
{
CV_TRACE_FUNCTION();
if (rootModule == NULL)
{
rootModule = new Module("Sequential");
curModule = rootModule;
CV_Assert(rootModule == NULL);
cv::Ptr<Module> rootModule_ = cv::makePtr<Module>("Sequential");
rootModule = rootModule_.get();
curModule = rootModule;
THFile_seek(file, 0);
readObject();
}
THFile_seek(file, 0);
readObject();
net = net_;
std::vector<std::pair<int, Module*> > addedModules;
fill(rootModule, addedModules);
rootModule = NULL;
curModule = NULL;
}
};