diff --git a/Logic/TrashRecognition/Retrain.py b/Logic/TrashRecognition/Retrain.py index 1d67571..ea8db80 100644 --- a/Logic/TrashRecognition/Retrain.py +++ b/Logic/TrashRecognition/Retrain.py @@ -1070,13 +1070,13 @@ if __name__ == '__main__': parser.add_argument( '--image_dir', type=str, - default='', + default='../../Images/TrainingImages', help='Path to folders of labeled images.' ) parser.add_argument( '--output_graph', type=str, - default='/tmp/output_graph.pb', + default='../../Model/retrained_graph.pb', help='Where to save the trained graph.' ) parser.add_argument( @@ -1097,13 +1097,13 @@ if __name__ == '__main__': parser.add_argument( '--output_labels', type=str, - default='/tmp/output_labels.txt', + default='../../Model/retrained_labels.txt', help='Where to save the trained graph\'s labels.' ) parser.add_argument( '--summaries_dir', type=str, - default='/tmp/retrain_logs', + default='/retrain_logs', help='Where to save summary logs for TensorBoard.' ) parser.add_argument( @@ -1235,7 +1235,7 @@ if __name__ == '__main__': parser.add_argument( '--saved_model_dir', type=str, - default='', + default='../../Model', help='Where to save the exported graph.') parser.add_argument( '--logging_verbosity', diff --git a/Model/retrained_graph.pb b/Model/retrained_graph.pb new file mode 100644 index 0000000..8044846 Binary files /dev/null and b/Model/retrained_graph.pb differ diff --git a/Model/retrained_labels.txt b/Model/retrained_labels.txt new file mode 100644 index 0000000..96012c6 --- /dev/null +++ b/Model/retrained_labels.txt @@ -0,0 +1,4 @@ +glass +metal +paper +plastic diff --git a/Model/saved_model.pb b/Model/saved_model.pb new file mode 100644 index 0000000..9cddf53 Binary files /dev/null and b/Model/saved_model.pb differ diff --git a/Model/variables/variables.data-00000-of-00001 b/Model/variables/variables.data-00000-of-00001 new file mode 100644 index 0000000..ef88ec0 Binary files /dev/null and b/Model/variables/variables.data-00000-of-00001 differ diff --git a/Model/variables/variables.index b/Model/variables/variables.index new file mode 100644 index 0000000..b656122 Binary files /dev/null and b/Model/variables/variables.index differ