mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Add TKDNN_CALIB_IMG_PATH and TKDNN_CALIB_LABRL_PATH variable

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+1 -2
View File
@@ -73,8 +73,7 @@ NetworkRT::NetworkRT(Network *net, const char *name) {
// builderRT->setInt8Mode(true);
configRT->setFlag(BuilderFlag::kINT8);
BatchStream calibrationStream(dim, 1, 100, //TODO: check if 100 images are sufficient to the calibration (or 4951)
"/home/xavier/Documents/tkDNN/demo/COCO_val2017/all_images.txt",
"/home/xavier/Documents/tkDNN/demo/COCO_val2017/all_labels.txt");
net->fileImgList, net->fileLabelList);
std::string modelName = name;
calibrator.reset(new Int8EntropyCalibrator(calibrationStream, 1,
"./" + modelName.substr(0, modelName.find('.')) + "-calibration.table",