tensorflow2.0和之前版本读取模型的接口变化

tensorflow2.0 c++加载模型可以用接口LoadSavedModel：

SessionOptions opts;

    tensorflow::RunOptions run_options;

    opts.config.mutable_gpu_options()->set_visible_device_list({m_deviceList.c_str()});

    Status status_create = LoadSavedModel(m_modelPath, {tensorflow::kSavedModelTagServe}, &m_bundle);

    if(!status_create.ok()){

        return false;

     }

 

tesorflow2.0之前的接口为ReadBinaryProto；

　GraphDef graph_def;

    Status status_load = ReadBinaryProto(Env::Default(), fullPath, &graph_def);       

    if (!status_load.ok()) {

        return false;

    }

 

模型也可以以字节流的方式输入模型：

GraphDef graph_def;
if (!graph_def.ParseFromArray(data_buff, readLen)) {
delete[]data_buff;
return FALSE;
}

SessionOptions opts;

opts.config.mutable_gpu_options()->set_allow_growth(true);

opts.config.mutable_gpu_options()->set_per_process_gpu_memory_fraction(0.5);

opts.config.mutable_gpu_options()->set_visible_device_list({"1,2,3,0"});

std::shared_ptr<tensorflow::Session> session_prt(tensorflow::NewSession(opts));

m_session_prt = session_prt;