update reqs

Signed-off-by: Niv Sardi <xaiki@evilgiggle.com>

python/write_data.py

@@ -1,5 +1,7 @@
 import csv
 import entity
+import argparse
+
 from common import defaults
 
 def gen_data_yaml(bcos):
@@ -13,5 +15,9 @@ names: [{names}]
 '''
 
 if __name__ == '__main__':
-    bcos =  entity.read_entities(defaults.MAIN_CSV_PATH)
+    parser = argparse.ArgumentParser(description='creates a YOLOv5 data.yaml')
+    parser.add_argument('csv', metavar='csv', type=str, 
+                    help='csv file')
+    args = parser.parse_args()
+    bcos =  entity.read_entities(args.csv)
     print(gen_data_yaml(bcos))