From 921da8c8b1a01ee0bc20f2d5466afbdf212b7916 Mon Sep 17 00:00:00 2001 From: HappyZ Date: Fri, 7 Jun 2019 10:29:28 -0500 Subject: [PATCH] Update localization_log_gamma_model.py --- localization_log_gamma_model.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/localization_log_gamma_model.py b/localization_log_gamma_model.py index e16ce51..1fbcc4e 100644 --- a/localization_log_gamma_model.py +++ b/localization_log_gamma_model.py @@ -32,11 +32,15 @@ def fittingSingle(filepath, args): print("err: failed to load file {}".format(filepath)) return location_gt_tmp = None - location_gt = [float('nan'), float('nan')] - if len(data) is 2: - data, location_gt_tmp = data - if location_gt_tmp is not None: - location_gt = [location_gt_tmp[0] * 0.1, location_gt_tmp[1] * 0.1] + if 'img_' in os.path.basename(filepath): + location_gt_tmp = os.path.basename(filepath).split('_') + location_gt = [float(location_gt_tmp[1]) + 3.2, float(location_gt_tmp[2]) + 3.2] + else: + location_gt = [float('nan'), float('nan')] + if len(data) is 2: + data, location_gt_tmp = data + if location_gt_tmp is not None: + location_gt = [location_gt_tmp[0] * 0.1, location_gt_tmp[1] * 0.1] if args.only_true_loc and location_gt_tmp is None: print("skipped as no groundtruth") return @@ -116,7 +120,7 @@ def main(args): filepaths.append(args.filepath) if args.folderpath: files = os.listdir(args.folderpath) - filepaths.extend(["{}/{}".format(args.folderpath, file) for file in files if '.pickle' in file]) + filepaths.extend(["{}/{}".format(args.folderpath, file) for file in files if '.pickle' in file and '_floormap.pickle' not in file]) # prepare if args.outputfile: