How I can use this tool with a .pb/.h5 model?

[BIRlz]

Excuse me, I want to use this tool to visualize my own trained vgg16 model. However, I only have a .pb and .h5 model instead of .npy or .cpkt. So How can I convert or change some parts of code to achieve my my goal?
I would appreciate it if you can help me!

[sankin1770]

搞定了没，我改成了在ckpt恢复跑 已经实现

[nengbunengzhengdianqi]

搞定了没，我改成了在ckpt恢复跑 已经实现

你好，请问需要做哪些修改吗

[xiaomiu311621]

搞定了没，我改成了在ckpt恢复跑 已经实现

请问怎么改成在恢复的ckpt模型实现呢？

[Doflamingo-swj]

搞定了没，我改成了在ckpt恢复跑 已经实现

你好！请问ckpt文件应该怎么加载进去呢，而不是h5或者其他存储的方式。