MiniDNN icon indicating copy to clipboard operation
MiniDNN copied to clipboard

Doing inference/prediction only, with weights loaded from a Tensorflow NN

Open josephernest opened this issue 4 years ago • 1 comments
trafficstars

Nice project! Is it possible to do the following:

  1. Train a CNN with, for example, Tensorflow + Python
  2. Save the model weights into a file
  3. Load this weights file from MiniDNN / C++ code, and do prediction/inference only with MiniDNN (no training done with MiniDNN)

How would it be possible to load weights/coefficients of a model pre-trained with Tensorflow, into MiniDNN?

josephernest avatar Jun 22 '21 21:06 josephernest

I'd also love to know this answer.

I haven't yet implemented it, but I've also found cppflow , which is a lightweight wrapper for tensorflow, if you're wanting to load a trained model into c++ code to draw inference from.

TimJones7 avatar Dec 31 '21 18:12 TimJones7