Mask_RCNN: как сохранить обученную модель и преобразовать ее в tflite

Question

Я следовал за уроком в https://machinelearningmastery.com/how-to-train-an-object-detection-model-with-keras/

После успешного обучения я получил 5 .h5 файлов:

mask_rcnn_kangaroo_cfg_0001.h5
mask_rcnn_kangaroo_cfg_0002.h5
mask_rcnn_kangaroo_cfg_0003.h5
mask_rcnn_kangaroo_cfg_0004.h5
mask_rcnn_kangaroo_cfg_0005.h5

Я новичок ie к этому, так что мое понимание может быть неправильным:

Как я могу преобразовать эти .h5 файлы в .pb или лучше в .tflite файлов, так что я могу использовать их в Android приложении обнаружения объектов?

Answer 1

Вам не нужно конвертировать эти .h5 в .pb, вы можете напрямую конвертировать файлы keras .h5 в tflite. Вот официальная документация по как .

Убедитесь, что модель со слоями поддерживается TFLite, как указано здесь .

Один раз у вас есть модель .tflite, которую вы можете запустить интерпретатором на Android.