s434720/2019SZI-Projekt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update graph, labels and image classification script

Browse Source
This commit is contained in:
nlitkowski 2019-06-09 22:23:39 +02:00
parent 01a9c6f790
commit ed36703151
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Logic/TrashRecognition/ImageClassification.py
View File

@ -74,7 +74,7 @@ def load_labels(label_file):
if __name__ == "__main__": if __name__ == "__main__":
model_file = "Model/retrained_graph.pb" model_file = "Model/graph.pb"
label_file = "Model/retrained_labels.txt" label_file = "Model/retrained_labels.txt"
input_height = 299 input_height = 299
input_width = 299 input_width = 299
@ -86,11 +86,11 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--graph", parser.add_argument("--graph",
default="Model/retrained_graph.pb", default="Model/graph.pb",
help="graph/model to be executed") help="graph/model to be executed")
parser.add_argument("--labels", parser.add_argument("--labels",
default="Model/retrained_labels.txt", default="Model/graph_labels.txt",
help="name of file containing labels") help="name of file containing labels")
parser.add_argument("--input_height", type=int, help="input height") parser.add_argument("--input_height", type=int, help="input height")
@ -102,7 +102,7 @@ if __name__ == "__main__":
parser.add_argument("--input_std", type=int, help="input std") parser.add_argument("--input_std", type=int, help="input std")
parser.add_argument("--input_layer", parser.add_argument("--input_layer",
default="Placeholder", default="Mul",
help="name of input layer") help="name of input layer")
parser.add_argument("--output_layer", parser.add_argument("--output_layer",

BIN
Model/graph.pb Normal file
View File

Binary file not shown.

4
Model/graph_labels.txt Normal file
View File

@ -0,0 +1,4 @@
glass
metal
paper
plastic