Updated retrain script. Add retrained graph.

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',

Model/retrained_labels.txt

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