Я работаю над программным обеспечением для определения номера. Я обучил API обнаружения объектов TensorFlow на заказном наборе данных номерных знаков в google сотрудники. После этого я скачал frozen_inference_graph.pb из папки обучения, что является конечным результатом после обучения. Изначально я использовал save_model.pb, который был загружен из концентратора TensorFlow, и он работает нормально, но теперь я переименовал этот frozen_graph_inference.pb в save_model.pb, и когда я его использовал, выдает ошибку
КОД, ИСПОЛЬЗУЕМЫЙ ДЛЯ ЗАГРУЗКА SAVED_MODEL
#picking an object detection module
module_handle = "Models /" детектор = hub.load (module_handle) .signatures ['default']
Здесь модели это каталог где я храню все модели. Когда я использую свою собственную модель, я получаю эту ошибку
Файл "objectDetection.py", строка 2, из импорта helperFunctions * Файл "C: \ MachineLearning \ helperFunctions.py", строка 89, в Detector = hub.load (module_handle) .signatures ['default'] Файл "C: \ MachineLearning \ env \ lib \ site-packages \ensorflowflow_core \ python \ save_model \ signature_serialization.py", строка 195, in getitem return self._signatures [key] KeyError: 'default'
Любая помощь будет принята с благодарностью. Спасибо