jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-20 01:27:06 -05:00
Code Issues Projects Releases Wiki Activity

update samples from Release-53 as a part of SDK release

Browse Source
This commit is contained in:
vizhur
2020-05-27 02:45:07 +00:00
parent 12c06a4168
commit 11e8ed2bab
74 changed files with 1085 additions and 713 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tutorials/machine-learning-pipelines-advanced/scripts/batch_scoring.py
View File

@@ -21,9 +21,10 @@ image_size = 299
num_channel = 3
def get_class_label_dict():
def get_class_label_dict(labels_dir):
label = []
proto_as_ascii_lines = tf.gfile.GFile("labels.txt").readlines()
labels_path = os.path.join(labels_dir, 'labels.txt')
proto_as_ascii_lines = tf.gfile.GFile(labels_path).readlines()
for l in proto_as_ascii_lines:
label.append(l.rstrip())
return label
@@ -34,14 +35,10 @@ def init():
parser = argparse.ArgumentParser(description="Start a tensorflow model serving")
parser.add_argument('--model_name', dest="model_name", required=True)
parser.add_argument('--labels_name', dest="labels_name", required=True)
parser.add_argument('--labels_dir', dest="labels_dir", required=True)
args, _ = parser.parse_known_args()
workspace = Run.get_context(allow_offline=False).experiment.workspace
label_ds = Dataset.get_by_name(workspace=workspace, name=args.labels_name)
label_ds.download(target_path='.', overwrite=True)
label_dict = get_class_label_dict()
label_dict = get_class_label_dict(args.labels_dir)
classes_num = len(label_dict)
with slim.arg_scope(inception_v3.inception_v3_arg_scope()):