[TensorRT] keras model 에서 node name 확인하는 방법

 

아래와 같은 에러 발생 시

 

[TensorRT] ERROR: Parameter check failed at: ../builder/Network.cpp::addInput::406, condition: isValidDims(dims)
[TensorRT] ERROR: UFFParser: Failed to parseInput for node input_1
[TensorRT] ERROR: UFFParser: Parser error: input_1: Failed to parse node - Invalid Tensor found at node input_1

 

 

 

input shape을  (224, 224, 3) 가 아닌 [224, 224, 3] 으로 넣어주어야 함 

그리고 keras model의 input node name, output node name 을 확인하여 올바르게 넣어줄 것 

 

keras model 에서 node name 확인하는 방법은 아래와 같다.

 

from keras.models import load_model

model = load_model("./model.h5")
model.summary()

# input
print([node.op.name for node in model.inputs])

# output
print([node.op.name for node in model.outputs])

 

 

 

참고자료

https://devtalk.nvidia.com/default/topic/1048762/tensorrt-5-0-converting-to-and-from-uff/

https://devtalk.nvidia.com/default/topic/1048762/tensorrt-5-0-converting-to-and-from-uff/