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

Fix file name to exclude from nbCalibFiles count

parent cca44cf0
No related branches found
No related tags found
No related merge requests found
......@@ -120,7 +120,7 @@ void Graph::datamode(nvinfer1::DataType datatype)
}
void Graph::calibrate()
{
// Open calbration files
// Open calibration files
const std::string calibDir = this->_calibPath;
std::vector<std::string> filesCalib;
struct dirent* pFile;
......@@ -140,8 +140,8 @@ void Graph::calibrate()
// Get input tensor shape by reading data.info file in calibration folder
std::vector<int> dims;
std::ifstream inputFile(calibDir + "/data.info");
if (!inputFile.is_open()) std::cout << "Error opening the file!" << std::endl;
std::ifstream inputFile(calibDir + "/.info");
if (!inputFile.is_open()) std::cout << "Error opening the file .info" << std::endl;
std::string line;
for (std::size_t i = 0; i < 3 && std::getline(inputFile, line); ++i) {
try {
......@@ -154,7 +154,7 @@ void Graph::calibrate()
}
}
inputFile.close();
std::cout << nbCalibFiles << " calibration files found" << std::endl;
BatchStream calibrationStream(1, dims[0], dims[1], dims[2], nbCalibFiles, this->_calibPath);
this->_calibrator = new Int8EntropyCalibrator(calibrationStream, 0, this->_cachePath);
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