Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable

This commit adds two variables for the calibration dataset.
The first is reffered to .txt file that contains the list
of the absolute paths of the images for the INT8 calitration.
The second is referred to .txt file that contains the list
of the absolute paths of the labels of the same images above.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-03-30 18:55:32 +02:00
parent ce20868bad
commit bb157be82c
3 changed files with 10 additions and 2 deletions
+6
View File
@@ -34,6 +34,12 @@ Network::Network(dataDim_t input_dim) {
int8 = true;
}
}
if(const char* env_p = std::getenv("TKDNN_CALIB_IMG_PATH"))
fileImgList = env_p;
if(const char* env_p = std::getenv("TKDNN_CALIB_LABEL_PATH"))
fileLabelList = env_p;
if(fp16)
std::cout<<COL_REDB<<"!! FP16 INFERENCE ENABLED !!"<<COL_END<<"\n";