happyz
/
xiaomi_vacuum_as_data_collector
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add orientation check in foldername

This commit is contained in:
HappyZ 2019-06-05 12:41:15 -05:00
parent c8f71d9a88
commit e9fafdbb8a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
preprocessor.py
View File

@ -119,6 +119,9 @@ def main(args):
if not os.path.isdir(args.folder): if not os.path.isdir(args.folder):
print("Err: folder {} does not exist".format(args.folder)) print("Err: folder {} does not exist".format(args.folder))
sys.exit(2) 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) f_map, f_loc, f_sig, f_gt, is_csi = get_files(args.folder)
if f_loc is None or f_sig is None: if f_loc is None or f_sig is None:
print("Err: desired files not exist") print("Err: desired files not exist")