Procházet zdrojové kódy

Add command line argument to csv printing script

Mario Hüttel před 2 roky
rodič
revize
6be539f738
1 změnil soubory, kde provedl 6 přidání a 1 odebrání
  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)})