C++テキストファイルを行ごとに読み込む
最近TensorRTを勉強して、1つの問題に出会って、自分で書いた層の中でいくつかのパラメータは関数を通じて伝来することができなくて、例えばRenfindet SSD層が必要とするNMS閾値と信頼度閾値はすべて中に書いて死んで、また外部に伝来することができません.インタフェースから入力されたパラメータをテキストファイルに保存し、テキストファイルの文字列を読み取ることで、しきい値をexternパラメータを使用してdetect_に入力します.out層、結果検出を行う;
次のコードはテキストファイルを解析し、NMSしきい値と信頼度しきい値を取得する.
補足:定義された構造体
構造体のインスタンス化:
次のコードはテキストファイルを解析し、NMSしきい値と信頼度しきい値を取得する.
float find_str(string str_src,string str_dest)
{
if (!str_src.empty())
{
int index = 0;
while( (index = str_src.find(' ',index)) != string::npos)
{
str_src.erase(index,1);
}
int end = str_src.find_first_not_of(str_dest);
if (end != -1){
// cout<
補足:定義された構造体
typedef struct Obj_Score_
{
float conf_score;
float nms_score;
}Obj_score;
構造体のインスタンス化:
Obj_score thresh;