Для запуска демо мне понадобилась библиотека GDAL и rasterio
1.Я взял за основу демо "fire_detection_demo" из открытого "pull requests", проверил, чтоб оно запускалось, и попутно собирало модель. После этого сохранил модель при помощи "model.save('my_h5_model.h5')" в самом конце.
2.После сохранения модели я закомментировал get_model и load_weights, и просто загрузил модель через load_model ('path to model'), прикрепил скриншот №1.
3.После того, как я сохранил модель, мне для конвертации понадобились библиотеки onnx, keras2onnx по примеру из гайда, оттуда же я взял код для конвертации: https://medium.com/analytics-vidhya/how-to-convert-your-keras-model-to-onnx-8d8b092c4e4f
Но данный код запускается с ошибкой:
Для запуска демо мне понадобилась библиотека GDAL и rasterio 1.Я взял за основу демо "fire_detection_demo" из открытого "pull requests", проверил, чтоб оно запускалось, и попутно собирало модель. После этого сохранил модель при помощи "model.save('my_h5_model.h5')" в самом конце. 2.После сохранения модели я закомментировал get_model и load_weights, и просто загрузил модель через load_model ('path to model'), прикрепил скриншот №1.
3.После того, как я сохранил модель, мне для конвертации понадобились библиотеки onnx, keras2onnx по примеру из гайда, оттуда же я взял код для конвертации: https://medium.com/analytics-vidhya/how-to-convert-your-keras-model-to-onnx-8d8b092c4e4f
Но данный код запускается с ошибкой:
