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

Update retrain script

Browse Source
This commit is contained in:
Norbert Litkowski 2019-05-06 03:15:57 +01:00
parent 0713f42692
commit eebeedcdff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Logic/TrashRecognition/Retrain.py
View File

@ -1175,7 +1175,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--output_graph', '--output_graph',
type=str, type=str,
default='Model/output_graph.pb', default='Model/retrained_graph.pb',
help='Where to save the trained graph.' help='Where to save the trained graph.'
) )
parser.add_argument( parser.add_argument(
@ -1196,7 +1196,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--output_labels', '--output_labels',
type=str, type=str,
default='Model/output_labels.txt', default='Model/retrained_labels.txt',
help='Where to save the trained graph\'s labels.' help='Where to save the trained graph\'s labels.'
) )
parser.add_argument( parser.add_argument(