Я пытаюсь преобразовать сохраненную_модель из модели обнаружения объектов TensorFlow Zoo в tflite файл. Я использую модель ssd_mobilenet_v2_oid_v4_2018_12_12 .
Поэтому я использовал export_tflite_ssd_graph.py :
D:\TensorFlow\models\research\object_detection>python export_tflite_ssd_graph.py
--pipeline_config_path=D:\Models\ssd_mobilenet_v2_oid_v4_2018_12_12\pipeline.config
--trained_checkpoint_prefix=D:\Models\ssd_mobilenet_v2_oid_v4_2018_12_12\model.ckpt.data-00000-of-00001
--output_directory=D:\Models
--add_postprocessing_op=true
Я получаю следующую ошибку
2020-01-20 19:43:36.765263: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
2020-01-20 19:43:36.770058: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
Traceback (most recent call last):
File "export_tflite_ssd_graph.py", line 96, in <module>
from object_detection import export_tflite_ssd_graph_lib
File "C:\Users\Bhavin\AppData\Local\Programs\Python\Python36\lib\site-packages\object_detection-0.1-py3.6.egg\object_detection\export_tflite_ssd_graph_lib.py", line 26, in <module>
from tensorflow.tools.graph_transforms import TransformGraph
ModuleNotFoundError: No module named 'tensorflow.tools.graph_transforms'
Я пытался обновить tenorflow с помощью pip, но все равно получаю ту же ошибку.
Я использую Windows 10 и TensorFlow 2.1.0.