Skip to content
Snippets Groups Projects
Commit 58fca115 authored by Nathan Thoumine's avatar Nathan Thoumine
Browse files

Fix default calibration folder / cache

parent 3d3527a7
No related branches found
No related tags found
No related merge requests found
......@@ -26,7 +26,7 @@ void init_Graph(py::module& m)
.def("device", &Graph::device, py::arg("id"))
.def("load", &Graph::load, py::arg("filepath"))
.def("save", &Graph::save, py::arg("filepath"))
.def("calibrate", &Graph::calibrate, py::arg("calibration_folder_path"), py::arg("cache_file_path"))
.def("calibrate", &Graph::calibrate, py::arg("calibration_folder_path") = "./calibration_folder/", py::arg("cache_file_path") = "./calibration_chache")
.def("initialize", &Graph::initialize)
.def("profile", &Graph::profile, py::arg("nb_iterations"), py::arg("mode")= ExecutionMode_T::ASYNC)
.def("run_sync", [](Graph& graph, py::list inputs) -> py::list {
......
......@@ -112,10 +112,11 @@ void Graph::datamode(nvinfer1::DataType datatype)
break;
}
}
void Graph::calibrate(std::string const& calibration_folder_path, std::string const& cache_file_path)
void Graph::calibrate( std::string const& calibration_folder_path = "./calibration_folder/",
std::string const& cache_file_path = "./calibration_chache")
{
// Open calibration files
const std::string calibDir = calibPath;
const std::string calibDir = calibration_folder_path;
std::vector<std::string> filesCalib;
struct dirent* pFile;
DIR* pDir = opendir(calibDir.c_str());
......@@ -149,8 +150,8 @@ void Graph::calibrate(std::string const& calibration_folder_path, std::string co
}
inputFile.close();
BatchStream calibrationStream(1, dims[0], dims[1], dims[2], nbCalibFiles, calibPath);
this->_calibrator = new Int8EntropyCalibrator(calibrationStream, 0, cachePath);
BatchStream calibrationStream(1, dims[0], dims[1], dims[2], nbCalibFiles, calibration_folder_path);
this->_calibrator = new Int8EntropyCalibrator(calibrationStream, 0, cache_file_path);
this->_builderconfig->setInt8Calibrator(this->_calibrator);
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment