Update localization_log_gamma_model.py

This commit is contained in:
HappyZ 2019-06-07 12:12:54 -05:00
parent 5b83cbfad3
commit 40612ad801
1 changed files with 1 additions and 1 deletions

View File

@ -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: