add orientation check in foldername

2019-06-05 12:41:15 -05:00 · 2019-06-05 12:41:15 -05:00 · e9fafdbb8a
parent c8f71d9a88
commit e9fafdbb8a
1 changed files with 3 additions and 0 deletions
--- a/preprocessor.py
+++ b/preprocessor.py
@ -119,6 +119,9 @@ def main(args):
    if not os.path.isdir(args.folder):
        print("Err: folder {} does not exist".format(args.folder))
        sys.exit(2)
+    if 'orient' in args.folder:
+        orientation = int(args.folder.rstrip('/').split('_')[-1])
+        print("orientation: {}".format(orientation))
    f_map, f_loc, f_sig, f_gt, is_csi = get_files(args.folder)
    if f_loc is None or f_sig is None:
        print("Err: desired files not exist")