บันทึกไฟล์ .pb และ .pbtxt จาก keras

ฉันต้องได้รับไฟล์ .pb และ .pbtxt จาก โมเดล Keras เพื่อใช้โมเดลที่ได้รับการฝึกแล้ว โมเดล Keras พร้อมโมดูล DNN ของ OpenCV ใน C ++

ฉันสามารถรับไฟล์ .pb ได้โดยทำ model.save('model',save_format='tf') แต่ฉันไม่พบวิธีรับไฟล์ .pbtxt


c++ opencv keras tensorflow
person Barabanga    schedule 09.06.2020    source แหล่งที่มา

คำตอบ (1)


arrow_upward
0
arrow_downward

ฉันกำลังใช้งานการตั้งค่าเดียวกันและประสบปัญหาเดียวกัน ดังนั้นนี่คือสิ่งที่ฉันพบ:

  1. ใน OpenCV หากคุณใช้ dnn::readNetFromTensorflow() คุณไม่จำเป็นต้องใช้ไฟล์ ".pbtxt" จริงๆ แค่ ".pb" เพียงอย่างเดียวก็เพียงพอแล้ว

  2. สิ่งที่คุณสร้างด้วยเมธอด keras model.save() คือ tensorflow SavedModel ซึ่งสร้างหลายไฟล์ (พร้อมกับ .pb) แต่ .pb ไม่ทำงานคนเดียวและ OpenCV ไม่ยอมรับรูปแบบ SavedModel ดังนั้นคุณต้องสร้างกราฟหยุดนิ่ง ข้อมูลเพิ่มเติมที่นี่ และวิธีการสร้างกราฟแช่แข็งสำหรับ โมเดลของคุณ

person Coucou436    schedule 13.06.2020