Я использовал
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
Нужна помощь
Если ссылка выше не работает, можете ли вы отправить мне электронное письмо, чтобы я мог поделиться им с вами, спасибо за помощь
Я могу скачать файлы. Мы проверим это и свяжемся с вами.
Ваша ошибка указывает на несоответствие версии 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? <-- кажется, что это не проблема после просмотра вашей картинки
О, нвм, я думаю, что вы имели в виду версию тензорного потока после установки тензорного потока, устанавливающего предварительные условия \ install_prerequisites_tf2.bat (версия тензорного потока 2.4.4 из этого файла). Однако как я могу обновить его до tensorflow 2.5.3 с поддержкой открытой версии vino 2021.4.2 LTS....
Вы можете установить с помощью этой команды: pip install tensorflow == 2.5.3 или вы можете сначала попробовать 2.4.4 и посмотреть, совместима ли эта версия.
Извините за позднее уведомление, у меня была огромная операция, поэтому я вернусь через неделю и дам вам знать, что я получил
Вы можете обратиться к Open Model Zoo Demos о том, как использовать XML и bin-файл.
Не могли бы вы поделиться файлами моделей .pb, чтобы мы могли их проверить с нашей стороны?