Commit 3afebbf8 authored by Cresson Remi's avatar Cresson Remi
Browse files

Merge branch '2-migration-tf2' of gitlab-ssh.irstea.fr:remi.cresson/otbtf into 2-migration-tf2

parents a5206850 dc3182d5
3 merge requests!15Develop,!13Migrate big files to git-lfs,!3Migration to TensorFlow2
Showing with 9 additions and 1 deletion
+9 -1
......@@ -83,6 +83,14 @@ void GetTensorAttributes(const tensorflow::protobuf::Map<std::string, tensorflow
dataTypes.clear();
dataTypes.reserve(tensorsNames.size());
itkDebugMacro("Nodes contained in the model: ");
int i = 0;
for (auto const & layer : layers)
{
itkDebugMacro("Node "<< i << " inside the model: " << layer.first);
i+=1;
}
// Get infos
for (std::vector<std::string>::iterator nameIt = tensorsNames.begin();
nameIt != tensorsNames.end(); ++nameIt)
......@@ -99,7 +107,7 @@ void GetTensorAttributes(const tensorflow::protobuf::Map<std::string, tensorflow
found = true;
const tensorflow::TensorInfo& tensor_info = layer.second;
itkDebugMacro("Found : " << layername << " in the model");
itkDebugMacro("Found: " << layername << " in the model");
// Set default to DT_FLOAT
tensorflow::DataType ts_dt = tensorflow::DT_FLOAT;
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment