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