Update main.py

Lexaul · Jun 24, 2019 · da26cc6 · da26cc6
1 parent 3dbe880
commit da26cc6
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/main/main.py b/main/main.py
@@ -29,7 +29,18 @@
 parser.add_argument('-i', '--input_dir', default='input', type=str, help='Path to input directory')
 parser.add_argument('-o', '--output_dir', default='output', type=str, help='Path to output directory')
 parser.add_argument('-t', '--thickness', default='1', type=int, help='Bounding box and cross line thickness')
-parser.add_argument('--tracker', default='DASIAMRPN', type=str, help='Type of tracker being used')
+parser = argparse.ArgumentParser(description='Open-source image labeling tool')
+parser.add_argument('-i', '--input_dir', default='input', type=str, help='Path to input directory')
+parser.add_argument('-o', '--output_dir', default='output', type=str, help='Path to output directory')
+parser.add_argument('-t', '--thickness', default='1', type=int, help='Bounding box and cross line thickness')
+'''
+tracker_types = ['CSRT', 'KCF','MOSSE', 'MIL', 'BOOSTING', 'MEDIANFLOW', 'TLD', 'GOTURN', 'DASIAMRPN']
+    Recomended tracker_type:
+        KCF -> KCF is usually very good (minimum OpenCV 3.1.0)
+        CSRT -> More accurate than KCF but slightly slower (minimum OpenCV 3.4.2)
+        MOSSE -> Less accurate than KCF but very fast (minimum OpenCV 3.4.1)
+'''
+parser.add_argument('--tracker', default='DASIAMRPN', type=str, help='tracker_type being used: ['CSRT', 'KCF','MOSSE', 'MIL', 'BOOSTING', 'MEDIANFLOW', 'TLD', 'GOTURN', 'DASIAMRPN']')
 parser.add_argument('-n', '--n_frames', default='50', type=int, help='number of frames to track object for')
 args = parser.parse_args()