Оптимизатор модели в Intel Open Vino

Я использовал

import tensorflow as tf
model = tf.keras.models.load_model('model.h5')
tf.saved_model.save(model,'model')

Для сохранения моей модели классификации изображений (версия tensorflow в google colab = 2.9.2, версия intel open vino [Инструменты разработки] = 2021.4.2 LTS)

---------------------------------------------------------------------------------------
C:\Program Files (x86)\Intel\openvino_2021.4.752\deployment_tools\model_optimizer>python mo_tf.py --saved_model_dir C:\Users\dchoi\CNNProejct_Only_saved_English\saved_model --input_shape [1,32,320,240,3] --output_dir C:\Users\dchoi\CNNproject_only_output_English\output_model
Model Optimizer arguments:
Common parameters:
        - Path to the Input Model:      None
        - Path for generated IR:        C:\Users\dchoi\CNNproject_only_output_English\output_model
        - IR output name:       saved_model
        - Log level:    ERROR
        - Batch:        Not specified, inherited from the model
        - Input layers:         Not specified, inherited from the model
        - Output layers:        Not specified, inherited from the model
        - Input shapes:         [1,32,320,240,3]
        - Mean values:  Not specified
        - Scale values:         Not specified
        - Scale factor:         Not specified
        - Precision of IR:      FP32
        - Enable fusing:        True
        - Enable grouped convolutions fusing:   True
        - Move mean values to preprocess section:       None
        - Reverse input channels:       False
TensorFlow specific parameters:
        - Input model in text protobuf format:  False
        - Path to model dump for TensorBoard:   None
        - List of shared libraries with TensorFlow custom layers implementation:        None
        - Update the configuration file with input/output node names:   None
        - Use configuration file used to generate the model with Object Detection API:  None
        - Use the config file:  None
        - Inference Engine found in:    C:\Users\dchoi\AppData\Local\Programs\Python\Python38\lib\site-packages\openvino
Inference Engine version:       2021.4.0-3839-cd81789d294-releases/2021/4
Model Optimizer version:        2021.4.2-3974-e2a469a3450-releases/2021/4
[ WARNING ] Model Optimizer and Inference Engine versions do no match.
[ WARNING ] Consider building the Inference Engine Python API from sources or reinstall OpenVINO (TM) toolkit using "pip install openvino==2021.4"
2022-11-19 01:34:44.207311: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2022-11-19 01:34:44.207542: 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.
C:\Users\dchoi\AppData\Roaming\Python\Python38\site-packages\tensorflow\python\autograph\impl\api.py:22: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses
  import imp
2022-11-19 01:34:46.961002: I tensorflow/compiler/jit/xla_cpu_device.cc:41] Not creating XLA devices, tf_xla_enable_xla_devices not set
2022-11-19 01:34:46.961949: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'nvcuda.dll'; dlerror: nvcuda.dll not found
2022-11-19 01:34:46.962904: W tensorflow/stream_executor/cuda/cuda_driver.cc:326] failed call to cuInit: UNKNOWN ERROR (303)
2022-11-19 01:34:46.969471: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:169] retrieving CUDA diagnostic information for host: DESKTOP-SCBPOUA
2022-11-19 01:34:46.969727: I tensorflow/stream_executor/cuda/cuda_diagnostics.cc:176] hostname: DESKTOP-SCBPOUA
2022-11-19 01:34:46.970663: I tensorflow/core/platform/cpu_feature_guard.cc:142] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations:  AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-11-19 01:34:46.971135: I tensorflow/compiler/jit/xla_gpu_device.cc:99] Not creating XLA devices, tf_xla_enable_xla_devices not set
[ FRAMEWORK ERROR ]  Cannot load input model: SavedModel format load failure: NodeDef mentions attr 'validate_shape' not in Op<name=AssignVariableOp; signature=resource:resource, value:dtype -> ; attr=dtype:type; is_stateful=true>; NodeDef: {{node AssignNewValue}}. (Check whether your GraphDef-interpreting binary is up to date with your GraphDef-generating binary.).
------------------------------------------------------------------------------------------

Я получаю такую ​​​​ошибку даже после того, как я скачал

install_prerequirement/install_prerequisites_tf2.bat

Нужна помощь

Не могли бы вы поделиться файлами моделей .pb, чтобы мы могли их проверить с нашей стороны?

Aznie_Intel 21.11.2022 04:40
drive.google.com/drive/folders/…
newbieLife 22.11.2022 05:13

Если ссылка выше не работает, можете ли вы отправить мне электронное письмо, чтобы я мог поделиться им с вами, спасибо за помощь

newbieLife 22.11.2022 05:14

Я могу скачать файлы. Мы проверим это и свяжемся с вами.

Aznie_Intel 22.11.2022 09:10
[JS за 1 час] - 9. Асинхронный
[JS за 1 час] - 9. Асинхронный
JavaScript является однопоточным, то есть он может обрабатывать только одну задачу за раз. Для обработки длительных задач, таких как сетевые запросы,...
Топ-10 компаний-разработчиков PHP
Топ-10 компаний-разработчиков PHP
Если вы ищете надежных разработчиков PHP рядом с вами, вот список лучших компаний по разработке PHP.
Скраппинг поиска Apple App Store с помощью Python
Скраппинг поиска Apple App Store с помощью Python
📌Примечание: В этой статье я покажу вам, как скрапировать поиск Apple App Store и получить точно такой же результат, как на Apple iMac, потому что...
Редкие достижения на Github ✨
Редкие достижения на Github ✨
Редкая коллекция доступна в профиле на GitHub ✨
Подъем в javascript
Подъем в javascript
Hoisting - это поведение в JavaScript, при котором переменные и объявления функций автоматически "перемещаются" в верхнюю часть соответствующих...
Улучшение генерации файлов Angular
Улучшение генерации файлов Angular
Angular - это фреймворк. Вы можете создать практически любое приложение без использования сторонних библиотек.
1
4
101
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Ваша ошибка указывает на несоответствие версии TensorFlow, используемой для загрузки файла GraphDef. Из моей репликации я могу генерировать файлы промежуточного представления (IR), используя версию TensorFlow 2.5.3. Вот полная используемая команда оптимизатора модели:

mo_tf.py --saved_model_dir <path_to_model\IMGC.h5_to_saved_model.pb> --input_shape [1,320,240,3] --output_dir <path_for_output_files>

Итак, я использовал TensorFlow в Google Colab (версия tensorflow по умолчанию = 2.9.2). Я попытался повторно сохранить свою модель с помощью tensorflow версии 2.5.2 в google colab после того, как установил версию tensorflow 2.5.3 в google colab. Однако это дает мне ошибку: TypeError: __init__() получил неожиданный аргумент ключевого слова «ось» после того, как я запустил эти два кода ниже: model = tf.keras.models.load_model tf.saved_model.save(model, 'model_fixed') возможно потому что я использовал tf.keras.utils.image_dataset_from_directory, который недоступен в tensorflow 2.5.3? <-- кажется, что это не проблема после просмотра вашей картинки

newbieLife 22.11.2022 16:47

О, нвм, я думаю, что вы имели в виду версию тензорного потока после установки тензорного потока, устанавливающего предварительные условия \ install_prerequisites_tf2.bat (версия тензорного потока 2.4.4 из этого файла). Однако как я могу обновить его до tensorflow 2.5.3 с поддержкой открытой версии vino 2021.4.2 LTS....

newbieLife 22.11.2022 17:25

Вы можете установить с помощью этой команды: pip install tensorflow == 2.5.3 или вы можете сначала попробовать 2.4.4 и посмотреть, совместима ли эта версия.

Aznie_Intel 24.11.2022 04:42

Извините за позднее уведомление, у меня была огромная операция, поэтому я вернусь через неделю и дам вам знать, что я получил

newbieLife 27.11.2022 06:30
drive.google.com/drive/folders/… Я получаю это, я вижу много W и I, но это означает, что это сработало правильно? Кроме того, вы можете показать мне, как использовать этот xml и bin с Intel Open Model Zoo? мы должны запускать их в open cv или с intel open vino (и т. д. аналогично тому, как мы использовали оптимизатор модели в файле openvino)
newbieLife 07.12.2022 12:34

Вы можете обратиться к Open Model Zoo Demos о том, как использовать XML и bin-файл.

Aznie_Intel 22.12.2022 03:32

Другие вопросы по теме