mhu/micetoilet-labelgen
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add command line argument to csv printing script

Browse Source
This commit is contained in:
Mario Hüttel 2021-12-27 21:43:47 +01:00
parent fdcc35174f
commit 6be539f738
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
print_labels_from_csv.py
View File

@ -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)})