def _load_model(self):
# swig autoconverts py3 str to C++ string but expect byte strings on
# py2. As filenames should always be byte strings convert it back to
# str on py3 for swig.
if not PY2:
self.rnn = clstm.load_net(self.fname)
else:
self.rnn = clstm.load_net(self.fname.encode('utf-8'))
def load_clstm(fname):
"""
Loads a CLSTM model in protobuf format and instantiates an object
implementing the kraken.lib.SeqRecognizer interface.
Args:
fname (unicode): Path to the protobuf file
Returns:
A SeqRecognizer object
"""
try:
import clstm
except ImportError:
raise KrakenInvalidModelException('No clstm module available')
try:
clstm.load_net(fname.encode('utf-8'))
except Exception as e:
raise KrakenInvalidModelException(str(e))
return ClstmSeqRecognizer(fname)
def load_clstm(fname):
"""
Loads a CLSTM model in protobuf format and instantiates an object
implementing the kraken.lib.SeqRecognizer interface.
Args:
fname (unicode): Path to the protobuf file
Returns:
A SeqRecognizer object
"""
try:
import clstm
except ImportError:
raise KrakenInvalidModelException('No clstm module available')
try:
clstm.load_net(fname.encode('utf-8'))
except Exception as e:
raise KrakenInvalidModelException(str(e))
return ClstmSeqRecognizer(fname)
def _load_model(self):
self.rnn = clstm.load_net(self.fname.encode('utf-8'))
def _load_model(self):
self.rnn = clstm.load_net(self.fname.encode('utf-8'))
def load_model(self):
self.rnn = clstm.load_net(self.fname.encode("utf-8"))
