質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

新規登録して質問してみよう
ただいま回答率
85.48%
Keras

Kerasは、TheanoやTensorFlow/CNTK対応のラッパーライブラリです。DeepLearningの数学的部分を短いコードでネットワークとして表現することが可能。DeepLearningの最新手法を迅速に試すことができます。

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

Q&A

0回答

377閲覧

kerasの学習データをtensorRTで使用するためにuffファイルに変換したいです。

oinari_san

総合スコア10

Keras

Kerasは、TheanoやTensorFlow/CNTK対応のラッパーライブラリです。DeepLearningの数学的部分を短いコードでネットワークとして表現することが可能。DeepLearningの最新手法を迅速に試すことができます。

Python 3.x

Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

0グッド

0クリップ

投稿2023/01/10 08:17

実現したいこと

Jetson Nanoを用いてkerasの学習モデルをtensorRTで使用できるように.uffに変換しようとしています。
現在、このサイトを参考に.pbファイルに変換することができました。
https://qiita.com/dcm_yamaya/items/dcb149684fb210b39e8c

後、uffファイルに変換するために以下のコードをJetson Nano上で実行したのですがうまくいきません。
解決策を教えていただきたいです。
よろしくお願いします。

試した変換コード

import uff uff.from_tensorflow_frozen_model('fronzen_model.pb', output_nodes=output_names, output_filename='model.uff')

発生している問題・エラーメッセージ

2023-01-10 16:06:28.660195: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.2 WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them. WARNING:tensorflow:From /usr/lib/python3.6/dist-packages/uff/converters/tensorflow/conversion_helpers.py:274: The name tf.gfile.GFile is deprecated. Please use tf.io.gfile.GFile instead. NOTE: UFF has been tested with TensorFlow 1.15.0. WARNING: The version of TensorFlow installed on this system is not guaranteed to work with UFF. UFF Version 0.6.9 === Automatically deduced input nodes === [name: "conv2d_input" op: "Placeholder" attr { key: "dtype" value { type: DT_FLOAT } } attr { key: "shape" value { shape { dim { size: -1 } dim { size: 32 } dim { size: 32 } dim { size: 3 } } } } ] ========================================= Traceback (most recent call last): File "pb_uff.py", line 2, in <module> uff.from_tensorflow_frozen_model('/home/maruken/Desktop/model/save/frozen_model.pb',output_nodes='/home/maruken/Desktop/model/save/frozen_model.metadata',output_filename='/home/maruken/Desktop/model/save/model.uff') File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 276, in from_tensorflow_frozen_model return from_tensorflow(graphdef, output_nodes, preprocessor, **kwargs) File "/usr/lib/python3.6/dist-packages/uff/converters/tensorflow/conversion_helpers.py", line 198, in from_tensorflow name, graphdef.node, debug_mode=debug_mode) TypeError: 'str' object does not support item assignment

補足情報(FW/ツールのバージョンなど)

Python : 3.6
tensorflow : 1.15.0
keras : 2.4.3

気になる質問をクリップする

クリップした質問は、後からいつでもMYページで確認できます。

またクリップした質問に回答があった際、通知やメールを受け取ることができます。

バッドをするには、ログインかつ

こちらの条件を満たす必要があります。

guest

あなたの回答

tips

太字

斜体

打ち消し線

見出し

引用テキストの挿入

コードの挿入

リンクの挿入

リストの挿入

番号リストの挿入

表の挿入

水平線の挿入

プレビュー

まだ回答がついていません

会員登録して回答してみよう

アカウントをお持ちの方は

15分調べてもわからないことは
teratailで質問しよう!

ただいまの回答率
85.48%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

関連した質問