MycroftAI
/
mycroft-precise
mirror of https://github.com/MycroftAI/mycroft-precise.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Rename Tensorflow to TensorFlow

pull/1/head
Matthew D. Scholefield 2018-02-23 15:23:20 -06:00
parent f9c5dafd54
commit 8ced324465
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
precise/network_runner.py
View File

@ -22,7 +22,7 @@ class Runner(metaclass=ABCMeta):
pass pass
class TensorflowRunner(Runner): class TensorFlowRunner(Runner):
def __init__(self, model_name: str): def __init__(self, model_name: str):
if model_name.endswith('.net'): if model_name.endswith('.net'):
print('Warning: ', model_name, 'looks like a Keras model.') print('Warning: ', model_name, 'looks like a Keras model.')
@ -83,7 +83,7 @@ class Listener:
def find_runner(model_name: str) -> Type[Runner]: def find_runner(model_name: str) -> Type[Runner]:
runners = { runners = {
'.net': KerasRunner, '.net': KerasRunner,
'.pb': TensorflowRunner '.pb': TensorFlowRunner
} }
ext = splitext(model_name)[-1] ext = splitext(model_name)[-1]
if ext not in runners: if ext not in runners:

2
precise/scripts/record.py
View File

@ -17,7 +17,7 @@ usage = '''
Run a model on microphone audio input Run a model on microphone audio input
:model str :model str
Either Keras (.net) or Tensorflow (.pb) model to run Either Keras (.net) or TensorFlow (.pb) model to run
:-c --chunk-size int 2048 :-c --chunk-size int 2048
Samples between inferences Samples between inferences

2
precise/scripts/stream.py
View File

@ -15,7 +15,7 @@ usage = '''
will be given via stdout as a float string, one per line will be given via stdout as a float string, one per line
:model_name str :model_name str
Keras or Tensorflow model to read from Keras or TensorFlow model to read from
... ...
''' '''