fixes #65, ignore '-s' argument of rqt

Author: MatthijsBurgh

image_recognition_tensorflow/src/image_recognition_tensorflow/retrain.py

@@ -5,6 +5,7 @@
 from __future__ import print_function
 
 import os
+import sys
 from argparse import Namespace
 
 import errno
@@ -45,4 +46,8 @@ def main(image_dir, model_dir, output_dir, steps, batch, architecture="inception
     tf.app.flags.FLAGS.random_scale = random_scale
     tf.app.flags.FLAGS.random_brightness = random_brightness
 
-    tf_main('')
+    # Do not pass on the '-s' argument of rqt
+    if "rqt" in sys.argv[0]:
+        sys.argv.pop(1)
+
+    tf_main('')