rei-forks/precise-lite-amd64aarch64
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Support pb files in precise-test

Browse Source
This commit is contained in:
Matthew D. Scholefield 2018-04-26 19:04:05 -05:00
parent ef96763327
commit 989e29bb83
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
precise/scripts/test.py
View File

@ -17,6 +17,7 @@ from collections import namedtuple
from prettyparse import create_parser
from precise.model import load_precise_model
from precise.network_runner import Listener
from precise.params import inject_params
from precise.train_data import TrainData
@ -24,7 +25,7 @@ usage = '''
Test a model against a dataset
:model str
Keras model file (.net) to test
Either Keras (.net) or TensorFlow (.pb) model to test
:-t --use-train
Evaluate training data instead of test data
@ -101,7 +102,7 @@ def main():
inputs, targets = train if args.use_train else test
filenames = sum(data.train_files if args.use_train else data.test_files, [])
predictions = load_precise_model(args.model).predict(inputs)
predictions = Listener.find_runner(args.model)(args.model).predict(inputs)
stats = calc_stats(filenames, targets, predictions)
print('Data:', data)