TensorRTを使用してEfficientB0が動作しない
概要
pythonでTensorRTを実装したのですが、
EfficientB0モデルのみが推論結果が上手く表示されません
(複数枚画像渡しているが、１枚分の結果しかリターンされない)
他にもB2,B5推論してるのですが結果はしっかり返ってきてます
ちなみに
同じ学習スクリプト、変換方法、推論スクリプト使用してます
同じ事象に合った方がいれば、回答頂けると幸いです
1 likes
pythonでTensorRTを実装したのですが、
EfficientB0モデルのみが推論結果が上手く表示されません
(複数枚画像渡しているが、１枚分の結果しかリターンされない)
他にもB2,B5推論してるのですが結果はしっかり返ってきてます
ちなみに
同じ学習スクリプト、変換方法、推論スクリプト使用してます
同じ事象に合った方がいれば、回答頂けると幸いです
onnx変換をする際に「--inputs」指定していないのが問題でした！
ただ、B0のみこの現象が起きている原因がいまだにわかっていません