Simpan file .pb dan .pbtxt dari keras

Saya perlu mendapatkan file .pb dan .pbtxt dari model Keras untuk menggunakan yang terlatih Model keras dengan modul DNN OpenCV di C++.

Saya bisa mendapatkan file .pb dengan melakukan model.save('model',save_format='tf') tetapi saya belum menemukan cara untuk mendapatkan file .pbtxt.


c++ opencv keras tensorflow
person Barabanga    schedule 09.06.2020    source sumber

Jawaban (1)


arrow_upward
0
arrow_downward

Saya sedang mengerjakan pengaturan yang sama dan mengalami masalah yang sama, jadi inilah yang saya temukan:

  1. Di OpenCV jika Anda menggunakan dnn::readNetFromTensorflow() maka Anda sebenarnya tidak memerlukan file ".pbtxt", ".pb" saja sudah cukup

  2. Apa yang Anda hasilkan dengan metode keras model.save() adalah tensorflow SavedModel yang menghasilkan banyak file (bersama dengan .pb), tetapi .pb tidak bekerja sendiri dan OpenCV tidak menerima format SavedModel. Jadi, Anda perlu membuat grafik beku. Info lebih lanjut di sini, dan solusi untuk menghasilkan grafik yang dibekukan untuk model Anda.

person Coucou436    schedule 13.06.2020