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:
@@ -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";
|
||||
|
||||
Reference in New Issue
Block a user