Протестируйте файл .trt с помощью tensorflow
Ниже output_saved_model_dir в этом каталоге у меня есть файл trt с именем final_model_gender_classification_gpu0_int8.trt
output_saved_model_dir='/home/cocoslabs/Downloads/age_gender_trt'
saved_model_loaded = tf.saved_model.load(output_saved_model_dir, tags=[tag_constants.SERVING])
Когда я запускаю приведенный выше сценарий, он показывает следующую ошибку:
File "test.py", line 7, in <module>
saved_model_loaded = tf.saved_model.load(output_saved_model_dir, tags=[tag_constants.SERVING])
File "/home/cocoslabs/deepstream_docker/venv/lib/python3.6/site-packages/tensorflow_core/python/saved_model/load.py", line 528, in load
return load_internal(export_dir, tags)
File "/home/cocoslabs/deepstream_docker/venv/lib/python3.6/site-packages/tensorflow_core/python/saved_model/load.py", line 537, in load_internal
saved_model_proto = loader_impl.parse_saved_model(export_dir)
File "/home/cocoslabs/deepstream_docker/venv/lib/python3.6/site-packages/tensorflow_core/python/saved_model/loader_impl.py", line 83, in parse_saved_model
constants.SAVED_MODEL_FILENAME_PB))
OSError: SavedModel file does not exist at: /home/cocoslabs/Downloads/age_gender_trt/{saved_model.pbtxt|saved_model.pb}
Из приведенной выше ошибки я понимаю, что tf.saved_model.load() принимает только файлы.pb или.pbtxt. Это правильно? Но в соответствии с этой ссылкой Загрузите и запустите тестовую модель.trt, как они сказали, функция tf.saved_model.load() примет файл.trt. Помогите мне исправить эту ошибку. Благодарю вас.