C++ 按行读入文本文件

最近在学习TensorRT,遇到一个问题,就是自己写的层中有些参数通过函数没法传入,如Renfindet SSD层需要的NMS阈值和置信度阈值都是写死在里面,无法又外界传入。所以想了个傻办法:将界面传入的参数保存得文本文件中,然后通过读取文本文件的字符串,将阈值使用extern 参数录入detect_out层,进行结果检测;

下面的代码是对文本文件进行解析,获取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<<"3:"<

补充:定义的结构体

typedef struct Obj_Score_
{
    float conf_score;
    float nms_score;
}Obj_score;

 对结构体实例化:

Obj_score thresh;

 

你可能感兴趣的:(C++学习,TensorRT)