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
+3
View File
@@ -1,6 +1,7 @@
#ifndef NETWORK_H
#define NETWORK_H
#include <string>
#include "utils.h"
namespace tk { namespace dnn {
@@ -62,6 +63,8 @@ public:
bool fp16, dla, int8;
bool dontLoadWeights;
std::string fileImgList;
std::string fileLabelList;
};
}}