From 40612ad80175d806521540e8b6806ec93d11c8bb Mon Sep 17 00:00:00 2001 From: HappyZ Date: Fri, 7 Jun 2019 12:12:54 -0500 Subject: [PATCH] Update localization_log_gamma_model.py --- localization_log_gamma_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/localization_log_gamma_model.py b/localization_log_gamma_model.py index 43b7637..b4ea53f 100644 --- a/localization_log_gamma_model.py +++ b/localization_log_gamma_model.py @@ -32,7 +32,7 @@ def fittingSingle(filepath, args): print("err: failed to load file {}".format(filepath)) return location_gt_tmp = None - if 'img_' in os.path.basename(filepath): + if 'img_' == os.path.basename(filepath)[:4]: 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: