TF_SessionRun with multiple outputs gives Segmentation Fault about hello_tf_c_api

[pauperonway]

@Xonxt need help...

[Xonxt]

@pauperonway could you please give a bit more detail on how you modify the loadModel() and what exactly does your model's output look like?

[pauperonway]

now the result is ok.
i modify a few lines like this:(FYI)

bool tf_image::TF_Model::loadModel( const std::string & path, const double gpu_memory_fraction, bool inferIO )
{
// create an empty status;
...
//get final_binary_output
std::vectorstd::string outputNames;
outputNames.clear();
std::string binary_output_name = "net/final_binary_output";
name = binary_output_name.c_str();
//inputNames.push_back(binary_output_name);
outputNames.push_back(name);
// get last:
while ( (op = TF_GraphNextOperation( m_pGraph, &pos )) != nullptr ) {
name = TF_OperationName( op );
}
outputNames.push_back(name);
setOutputs( { outputNames } );
}
return true;
}

std::vectorcv::Mat tf_image::TF_Model::predict_image2image( const std::vectorcv::Mat& input )
{
...
// if everything ok, parse the output:
if ( !output_tensors.empty() ) {
// construct the method output:
for ( size_t i = 0; i < output_tensors.size(); i++ )
{
auto shape = GetTensorShape( output_tensors.at( i ) );
if (shape[2] <= 0 || shape[2] > 3) shape[2] = 1;
auto data = tf_utils::GetTensorsData( { output_tensors.at( i ) } );
// go through all outputs
output.push_back( cv::Mat( static_cast(shape[0]), static_cast(shape[1]), CV_32FC( static_cast(shape[2]) ), (void*) data[0].data() ) );
}
}
return output;
}

[pauperonway]

@Xonxt the key modify should be data[i] ---> data[0].

[pauperonway]

@Neargye hope this branch could merge to master.