Ver Fonte

Add command line argument to csv printing script

Mario Hüttel há 2 anos atrás
pai
commit
6be539f738
1 ficheiros alterados com 6 adições e 1 exclusões
  1. 6 1
      print_labels_from_csv.py

+ 6 - 1
print_labels_from_csv.py

@@ -2,9 +2,14 @@ import pandas as pd
 import shimatta_label.label_image as li
 import shimatta_label.brother_ql_wrapper as ql_wrapper
 import os
+import sys
 
 medir = os.path.dirname(__file__)
-example_data_path = os.path.join(medir, 'labels.csv')
+
+if len(sys.argv) > 2:
+    example_data_path = sys.argv[1]
+else:    
+    example_data_path = os.path.join(medir, 'labels.csv')
 
 df = pd.read_csv(example_data_path, converters={i: str for i in range(3)})