Giters

datature / portal

Portal is the fastest way to load and visualize your deep neural networks on images and videos 🔮

Home Page:https://datature.io/portal

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG]load tensorflow model fail

fanweiya opened this issue · comments

commented

I use YOLOv5 export tesnroflow save_model,use portal load this save model remind this model not a tensorflow model.

commented

Hi @fanweiya, thank you for raising this issue! I believe it might be because you have exported the model as a .h5 format instead of a saved_model folder containing a saved_model.pb. Currently we only support saved_model format instead of .h5 format.

commented

I have TF2 models trained on Google Cloud Platform and exported as saved_model.pb, but Portal fails to load it, and it returns the message "Are you sure this is a Tensorflow Model?". Indeed I am, and I use that for inference and it works just fine.

Perhaps I had to mention I am running portal as Windows Desktop Application under Windows 10 Pro.