Mario Hüttel пре 2 година
родитељ
комит
796f1da1e4
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      print_labels_from_csv.py

+ 2 - 2
print_labels_from_csv.py

@@ -6,7 +6,7 @@ import sys
 
 medir = os.path.dirname(__file__)
 
-if len(sys.argv) > 2:
+if len(sys.argv) > 1:
     example_data_path = sys.argv[1]
 else:    
     example_data_path = os.path.join(medir, 'labels.csv')
@@ -23,4 +23,4 @@ for _, row in df.iterrows():
     if row['Cut'] == 1:
         cut = True
     printer.print_image(file_path, cut = cut, rotation=270)
-    os.remove(file_path)
+    os.remove(file_path)